Win10环境下安装PyTorch GPU版【CUDA8.0+CUDNN8.0】

ゞ浴缸里的玫瑰 2022-03-16 06:22 525阅读 0赞

承接上篇博文win10环境下[《深度学习之CUDA与CUDNN详细安装教程》][CUDA_CUDNN][https://blog.csdn.net/Andone\_hsx/article/details/87942968][CUDA_CUDNN]

本文主要详细介绍PyTorch的安装过程和各种问题解决

PyTorch官网：[https://pytorch.org/][https_pytorch.org]

在安装PyTorch之前，我们需要安装好Anaconda:[https://www.anaconda.com/distribution/][https_www.anaconda.com_distribution]

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L0FuZG9uZV9oc3g_size_16_color_FFFFFF_t_70][]

双击下载并安装，注：选择默认模式，一直next安装！

安装好Anaconda后，打开PyTorch官网，选择相关对应的PyTorch版本

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L0FuZG9uZV9oc3g_size_16_color_FFFFFF_t_70 1][]

进入cmd， 复制并运行相应版本PyTorch下载安装执行语句

(上图红框内的语句:conda install pytorch torchvision cudatoolkit=8.0 -c pytorch)

等待安装......

安装过程中会出现，下载慢，下载不成功，即使设置了清华镜像，还是下到一半断掉

# CondaError: CondaHTTPError: HTTP 000 CONNECTION FAILED #

可通过选择设置**清华镜像+翻墙**来解决此问题。

设置清华镜像

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/  
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 --set show\_channel\_urls yes  
在Anaconda中按行执行，如果第一次设置，不会出现任何提示！

在翻墙的情况下，重新执行conda install pytorch torchvision cudatoolkit=8.0 -c pytorch即可！

测试PyTorch是否安装成功：

进入python IDE下

![20190226203134182.png][]

分别执行：

1、import torch

2、torch.\_\_version\_\_

3、torch.cuda.is\_available()

![20190226203734266.png][]

注意：torch.cuda.is\_available()是个函数，输出True，则表示cuda、cudnn、PyTorch均安装成功！

建议在使用Anaconda时创建虚拟环境，在虚拟环境下安装PyTorch！

相关博文：

[Anaconda创建虚拟环境][Anaconda]

[【Win10环境】深度学习之CUDA与CUDNN详细安装教程][CUDA_CUDNN]

[CUDA_CUDNN]: https://blog.csdn.net/Andone_hsx/article/details/87942968
[https_pytorch.org]: https://pytorch.org/
[https_www.anaconda.com_distribution]: https://www.anaconda.com/distribution/
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L0FuZG9uZV9oc3g_size_16_color_FFFFFF_t_70]: /images/20220316/d6c14f276fcf4cdab73f4ce838cf51bf.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L0FuZG9uZV9oc3g_size_16_color_FFFFFF_t_70 1]: /images/20220316/6f9dcc9d708c42f7a925b9cffe7baeae.png
[20190226203134182.png]: /images/20220316/7f6b8663027e43a3b564124197d9b206.png
[20190226203734266.png]: /images/20220316/c8144226bdf44ba3be61751545a860c0.png
[Anaconda]: https://blog.csdn.net/Andone_hsx/article/details/87948764