【记录-自用】Anoconda安装Pytorch-GPU踩坑点

前提：已经安装了Anoconda

1 下载和安装CUDA和Cudnn

第一次装，参考这篇博客: 下载和安装CUDA和Cudnn（图文详解）
第二次装，想先卸载以前的cuda，参考这篇博客：windows下CUDA的卸载以及安装

踩坑点：

• 官网下载Cudnn需要注册账号后才能下载，建议外网注册。

2 安装Pytorch-GPU

Pytorch官网安装方法失败，去掉"-c pytorch"安装出来的pytorch和torchvision版本对应失败。
我的方法： 离线下载pytorch、torchvision 点击这里下载，下载时注意cuda、python的版本，使用命令"pip debug --verbose"查看python，再执行pip install指令。一定要是对应版本，见踩坑点最后一点。

【PS】:https://download.pytorch.org/whl/torch_stable.html这个网站有时候打不开，真奇怪。

踩坑点：

• torch命令行执行pip install 不要用镜像下载，否则仍然是cpu版本。
• 如果已经安装了cpu版本的pytorch，建议重新创建一个虚拟环境安装gpu版本的。
• pytorch、torchvision 对应版本：Github网站

3 验证是否安装成功

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