安装与电脑CUDA匹配的pytorch版本

青旅半醒 2022-12-23 00:56 198阅读 0赞

# 查看是否支持以及版本号 #

[查看windows的GPU及CUDA版本][windows_GPU_CUDA]

# 卸载pytorch #

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

# 使用pip安装（未成功） #

## 官网生成命令 ##

[安装pytorch][pytorch]

![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L0dhYnJpZWxfd2Vp_size_16_color_FFFFFF_t_70_pic_center]  
官网命令

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`

![在这里插入图片描述][20201121151031908.png_pic_center]

为了防止各种问题我们 更改一下，[参考][Link 1]

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

![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L0dhYnJpZWxfd2Vp_size_16_color_FFFFFF_t_70_pic_center 1]

# 使用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`  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L0dhYnJpZWxfd2Vp_size_16_color_FFFFFF_t_70_pic_center 2]

# 测试环境是否安装成功 #

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

![在这里插入图片描述][20201121162825663.png_pic_center]

[windows_GPU_CUDA]: https://blog.csdn.net/Gabriel_wei/article/details/109896850
[pytorch]: https://pytorch.org/get-started/locally/
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L0dhYnJpZWxfd2Vp_size_16_color_FFFFFF_t_70_pic_center]: /images/20221120/731bd418e8654c069c5c28a058504467.png
[20201121151031908.png_pic_center]: /images/20221120/8a069fb5dbc249b9b00fcdb8d49a1d45.png
[Link 1]: https://www.cnblogs.com/xiximayou/p/13760560.html
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L0dhYnJpZWxfd2Vp_size_16_color_FFFFFF_t_70_pic_center 1]: /images/20221120/091bd76cc05f496685388a2ca8054357.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L0dhYnJpZWxfd2Vp_size_16_color_FFFFFF_t_70_pic_center 2]: /images/20221120/64b0201aa5fc4bd7bcf03123884dab14.png
[20201121162825663.png_pic_center]: /images/20221120/a10e68430c3e4312a5c85074c7f9d319.png