查看是否支持以及版本号

查看windows的GPU及CUDA版本

卸载pytorch

如果按的与版本不匹配,还需要卸载了
pip uninstall torch

使用pip安装(未成功)

官网生成命令

安装pytorch

在这里插入图片描述
官网命令

pip install torch===1.7.0 torchvision===0.8.1 torchaudio===0.7.0 -f https://download.pytorch.org/whl/torch_stable.html

查看pip当前源:

pip config list

在这里插入图片描述

为了防止各种问题我们 更改一下,参考

pip --default-timeout=1000 install --index-url https://pypi.tuna.tsinghua.edu.cn/simple/ torch===1.7.0 torchvision===0.8.1 torchaudio===0.7.0 -f https://download.pytorch.org/whl/torch_stable.html

在这里插入图片描述

使用Anaconda 安装(成功)

参考

https://blog.csdn.net/watermelon1123/article/details/88122020

添加清华源

该方案就是将官网的下载路径,改到清华镜像源的路径,先用activate激活自己之前创建的环境后,在命令行输入以下代码:

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源:

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/bioconda/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/menpo/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/

设置搜索时显示通道地址

conda config --set show_channel_urls yes

然后再去pytorch官网复制下载代码,记得删掉后面的-c pytorch,因为它表示从官网进行下载,不从清华镜像源下。

根据Python和CUDA选择对应的版本,然后官方给出提示可通过运行:

conda install pytorch torchvision torchaudio cudatoolkit=10.1 -c pytorch
conda install pytorch torchvision torchaudio cudatoolkit=10.1 

但是这里一定要注意,去掉-c pytorch,安装的时候才会默认从清华源下载相应的包,因此这里用命令行:

查看anaconda配置

conda config --show-sources
在这里插入图片描述

测试环境是否安装成功

import  torch
print (torch.cuda.is_available())

在这里插入图片描述

Logo

汇聚全球AI编程工具,助力开发者即刻编程。

更多推荐