环境操作相关命令
查看所有环境
conda env list
创建环境
conda create --name cahttts python=3.10
激活环境
conda activate cahttts
安装依赖文件
pip install -r requirements.txt
查看GPU型号
nvidia-smi -L
GPU 0: NVIDIA A10 (UUID: GPU-9f1fc9cf-582a-25ac-849c-2f77343f9a37)
删除虚拟环境
conda remove -n you_name --all
验证cuda环境
import torchif_cuda = torch.cuda.is_available()
print("if_cuda=", if_cuda)
gpu_count = torch.cuda.device_count()
print("gpu_count=", gpu_count)
无cuda环境
已安装conda环境
安装环境(依照自己的N卡coda版本来选择,ps:不要问,A卡应该怎么安装)
按照官网来选择安装
https://pytorch.org/
例如我现在的是Windows环境下的Conda安装,选择执行如下命令
conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia
查看torch版本
import torch
print(torch.__version__) #注意是双下划线
得到以下输出
2.1.2+cu118
修改conda环境的名称
修改conda环境名称(找到指定目录,直接修改文件夹即可)
conda相关命令
NVIDIA CUDA各版本下载链接
所有版本下载
https://developer.nvidia.com/cuda-toolkit-archive
11.8 cuda
https://developer.nvidia.com/cuda-11-8-0-download-archive