pytorch-gpu版本安装教程(亲测有效)

本教程是在annoconda下安装pytorch-gpu版本

步骤1：先根据英伟达的显卡的版本下载cuda

步骤2：然后下载对应版本的 cudnn  注意pytorch-cudnn-cuda  这个几个版本要对应的 所以下载之前要提前好看 装哪个cudnn

步骤3：装pytorch  直接用官网 PyTorch 的来安装有问题  下载很会慢

 这选择用清华镜像安装  网址： Index of /anaconda/cloud/pytorch/win-64/ | 清华大学开源软件镜像站 | Tsinghua Open Source Mirror

我这边选用了这个文件名下载的pytorch-1.7.1-py3.8_cuda110_cudnn8_0.tar.bz2

上面的文件名：python对应3.8版本  cuda对应11.0版本  cudnn对应8.0版本   上述的包只是下载了一个pytorch

步骤4：需要一个在你配置的环境终端那里输入conda install torchvision的指令 里面包含很多东西  比如：numpy包

步骤5：接着在配置的环境终端那里输入pip install spyder，下载我们的开发环境spyder

至此整个环境就配置好了

现在，测试你的pytorch能不能用gpu：

import torch

print(torch.cuda.is_available() )

安装过程中遇到过的问题 ： 报出的错误是：import win32api 这里无法打开

最终解决发现原因是因为 pywin32的版本问题 ，Pywin32==302 不能和python3.8匹配  需要pywin32==227 才行的