一行代码搞定cuda+cudnn虚拟环境创建

conda create -n torch python=3.12 cuda-toolkit=12 cudnn

 

cuda版本可以根据nvidia-smi查看，conda会自动配好各个模块的版本依赖。

 

cuda版的torch还是需要单独下载，conda直接从清华云下载安装的有问题。

可以从阿里云镜像下载安装：

pip install torch==2.5.1 torchvision torchaudio -f https://mirrors.aliyun.com/pytorch-wheels/cu124

 

.condarc换国内源下载快：

channels:
- https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
- https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
- https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
- https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
- https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge

- https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/fastai/
- https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
- https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/bioconda/
channel_priority: flexible
show_channel_urls: true