PyTorch官网按提示选择,分别是版本,操作系统,Conda下安装,CUDA10以上注意显卡驱动
但是在复制命令后,发现进度基本不动甚至还会出现网络错误的问题
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
这样做解决问题
最有效方法是添加镜像源,常见的清华或中科大
先查看是否已经安装相关镜像源,windows系统在 cmd
窗口中执行命令:
conda config --show
这里显示:
channels:
- https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
- http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
- http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
说明已经安装好清华的镜像源。如果没有安装,请参考下面命令安装源:
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
依次安装上面所有的源。
并设置搜索时显示通道地址,执行下面命令:
conda config --set show_channel_urls yes
重点
在安装完镜像源之后,安装代码记得去掉 -c pytorch 因为该命令默认还是从conda源下载。
正确命令:
conda install pytorch torchvision torchaudio cudatoolkit=10.2
顺利完成。
验证
在安装的环境中,输入python,然后返回 True 表示成功
>>>import torch
>>>torch.cuda.is_available()
True