分别输入下面四行代码(使用清华镜像):
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/
根据官网选择对应的pytorch版本:
https://pytorch.org/
其中根据控制面板查找电脑上cuda的版本
如:我使用的是CUDA=11.0,Windows下语言为python,在anaconda上进行安装,命令行如下:
conda install pytorch torchvision torchaudio cudatoolkit=11.0 -c pytorch
等待下载完成
遇到的问题:
1、过程中可能遇到无法下载完等问题,重新上述步骤即可。
2、下载完成后显示错误,提醒用户需要删除或重命名某些文件。此时关闭窗口,重新步骤即可;如无效可依照提示将需要删除的文件删除,再重复步骤。