安装 ：win10 + CUDA 11.0 + cuDNN8.0.2 + Pytorch-gpu

安装 ：win10 + CUDA 11.0 + cuDNN8.0.2 + Pytorch-gpu

前言

注意： 点进来的各位，你以为这是一篇正常的教程篇吗？

​ 不是的！ 这是一篇劝退文 。
安装方法在文末。

更新记录:
**1. 2020.10.15文末更新， 使用conda install pytorch torchvision cudatoolkit=11 -c pytorch-nightly 指令安装nightly版本 **
2. 2020.11.05: pytorch官网已经给出linux/windows/mac等的多源Stable版本安装方式，地址：https://pytorch.org/get-started/locally/#mac-installation

3. 2020.11.17: 测试了官网给出的cuda11的pytorch稳定版安装方法windows/pip

笔者安装cuda11+pytorch-gpu失败，故有此文。

安装时间：2020-08-07

原因

1. 安装CUDA和cuDNN时，请先打开Pytorch-gpu安装命令链接查看Pytorch是否有发布支持cuda11.0的binaries版本。

结果如下图：

可以看到，此时并无支持cuda11的binaries版本。
Pytorch论坛中给出了答案：

https://discuss.pytorch.org/t/pytorch-with-cuda-11-compatibility/89254

Pytorch源里贴出了windows 支持cuda11的PR：

https://github.com/pytorch/builder/pull/463

文末老哥劝一句，目前还是安装cuda10.2吧，从源build真的烦。

附cuda和cudnn链接：

cuDNN: https://developer.nvidia.com/rdp/cudnn-download
CUDA10.2 : https://developer.nvidia.com/cuda-10.2-download-archive

———————— 2020. 10. 15 更新 ————————

由博友帐下幕僚指出，可通过conda install指令的形式安装 cuda11&cudnn8.0的pytorch nightly版本，指令如下：

conda install pytorch torchvision cudatoolkit=11 -c pytorch-nightly

笔者在conda内测试过，确实可行，只是还没有实际安装不知道有没有什么坑要走。
等有时间了再测试一下，在此处更新。

2020.11.17更新 cuda11.0 pytorch安装方法

首先还是进入Pytorch-gpu安装命令链接查看对应环境的安装指令。

笔者测试环境是：

windows 10
conda python=3,7

切换到conda虚拟环境，使用pip安装指令：

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

安装结果：

C:\Users\Lenovo>activate ttt

(ttt) C:\Users\Lenovo>pip install torch===1.7.0+cu110 torchvision===0.8.1+cu110 torchaudio===0.7.0 -f https://download.pytorch.org/whl/torch_stable.html
Looking in indexes: http://pypi.douban.com/simple/
Looking in links: https://download.pytorch.org/whl/torch_stable.html
Collecting torch===1.7.0+cu110
  Downloading https://download.pytorch.org/whl/cu110/torch-1.7.0%2Bcu110-cp37-cp37m-win_amd64.whl (2046.8 MB)
     |████████████████████████████████| 2046.8 MB 123 bytes/s
Collecting torchvision===0.8.1+cu110
  Downloading https://download.pytorch.org/whl/cu110/torchvision-0.8.1%2Bcu110-cp37-cp37m-win_amd64.whl (1.6 MB)
     |████████████████████████████████| 1.6 MB 469 kB/s
Collecting torchaudio===0.7.0
  Downloading https://download.pytorch.org/whl/torchaudio-0.7.0-cp37-none-win_amd64.whl (103 kB)
     |████████████████████████████████| 103 kB 1.1 MB/s
Requirement already satisfied: numpy in e:\conda\envs\ttt\lib\site-packages (from torch===1.7.0+cu110) (1.19.4)
Collecting dataclasses
  Downloading http://pypi.doubanio.com/packages/26/2f/1095cdc2868052dd1e64520f7c0d5c8c550ad297e944e641dbf1ffbb9a5d/dataclasses-0.6-py3-none-any.whl (14 kB)
Processing c:\users\lenovo\appdata\local\pip\cache\wheels\71\78\b2\33e5003d632221538166dac06ffa89f842ff26e60400097ee3\future-0.18.2-py3-none-any.whl
Collecting typing-extensions
  Downloading http://pypi.doubanio.com/packages/60/7a/e881b5abb54db0e6e671ab088d079c57ce54e8a01a3ca443f561ccadb37e/typing_extensions-3.7.4.3-py3-none-any.whl (22 kB)
Collecting pillow>=4.1.1
  Downloading http://pypi.doubanio.com/packages/2b/65/e4a5130b4162d20ed99ff096549a04d18f050cfcdb16fe1643ac751c0181/Pillow-8.0.1-cp37-cp37m-win_amd64.whl (2.1 MB)
     |████████████████████████████████| 2.1 MB 939 kB/s
Installing collected packages: dataclasses, future, typing-extensions, torch, pillow, torchvision, torchaudio
Successfully installed dataclasses-0.6 future-0.18.2 pillow-8.0.1 torch-1.7.0+cu110 torchaudio-0.7.0 torchvision-0.8.1+cu110 typing-extensions-3.7.4.3

第一个1.9G 的whl文件下了一小会儿，中间会卡住一下变得很慢，不要慌，吃点东西喝个下午茶，结果就下好了。。

测试一下：

(ttt) C:\Users\Lenovo>pip list
Package           Version
----------------- -------------------
certifi           2020.11.8
dataclasses       0.6
future            0.18.2
numpy             1.19.4
pandas            1.1.4
Pillow            8.0.1
pip               20.2.4
pyperclip         1.8.1
python-dateutil   2.8.1
pytz              2020.4
setuptools        49.6.0.post20201009
six               1.15.0
torch             1.7.0+cu110
torchaudio        0.7.0
torchvision       0.8.1+cu110
typing-extensions 3.7.4.3
wheel             0.35.1
wincertstore      0.2
xlrd              1.2.0
XlsxWriter        1.3.7

(ttt) C:\Users\Lenovo>python
Python 3.7.8 | packaged by conda-forge | (default, Jul 31 2020, 01:53:57) [MSC v.1916 64 bit (AMD64)] on win32
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
>>>
>>> torch.__version__
'1.7.0+cu110'
>>>

测试结束，若安装有问题，欢迎评论区指出 ：)