在anaconda下载的话无需额外在官网下载cuda和cuDNN,annaconda会自动下载。
1.查看自己电脑的支持的CUDA版本。命令行nvidia-smi查看。
2.打开Start Locally | PyTorch,选择需要的版本,CUDA版本需要小于等于电脑支持的版本。
3.复制第二步给的命令。
4.打开anaconda终端,创建一个虚拟环境。conda create -n 环境名字 python=版本
conda create -n test python=3.8
5. 将当前虚拟环境切换到刚才创建的环境下。
conda activate test
6.添加镜像源(加快下载速度)
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors