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



