1.首先是配置清华镜像源
记事本打开 .condarc 文件(C:\Users\xxx),重写其中的内容。注意https改为http
channels:
- http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
- defaults
show_channel_urls: true
default_channels:
- http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
- http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/r
- http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/msys2
custom_channels:
conda-forge: http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
msys2: http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
bioconda: http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
menpo: http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
pytorch: http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
simpleitk: http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
2.接着在Anaconda prompt 命令窗口运行 conda clean -i 清除索引缓存,保证用的是镜像站提供的索引。
3.然后就可以安装 PyTorch啦
进入pytorch官网:https://pytorch.org/
网页下拉,即可看到下图,官网会自动根据你的电脑,显示的即是你可安装的CUDA版本,并给出安装命令。
复制官网提供的命令,并删除-c pytorch
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
打开 Anaconda prompt 命令窗口,进入你刚刚所创建的环境(我的命名是PyTorch)
conda activate PyTorch
进入环境后输入刚才复制好的命令进行下载conda install pytorch torchvision torchaudio cudatoolkit=10.2
注意:在测试过程中遇到cuda与pytorch版本不匹配的问题,我是选择下载一个旧一些的pytorch进行匹配,也可以对cuda进行更新。
4.测试
打开 Anaconda prompt 命令窗口,激活环境,输入python,进入python开发环境中
输入import torch
torch.cuda.is_available()
输出True就代表成功了