参考文章:
注意:在anaconda中使用conda命令安装pytorch时,不需要安装CUDA和对应的cudnn(参考文章2)
- 安装anaconda,官网直接安装即可
- 安装PyTorch,在官网找下载命令,然后到虚拟环境中输入即可。
- 打开Anaconda Prompt(开始菜单栏里面)
- 输入
conda create -n ml python=3.6
创建虚拟环境。ml作为自己的环境名 - 在虚拟环境中安装PyTorch,CUDA-10.2 PyTorch builds are no longer available for Windows, please use CUDA-11.3
- 使用清华源镜像安装pytorch
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
添加一次即可,如果有其他的镜像源冲突可以将其他的删去(我刚开始还添加了上交的,显示错误,删除就ok了)
然后根据官网的命令安装,注意将后面的
-c pytorch
删去,如果不删去的话意思是使用国外的源进行下载,速度奇慢。
conda install pytorch torchvision torchaudio cudatoolkit=11.3
第一次写文章,如有不足请各位前辈指出。