关于服务器使用pip 安装 pytorch安装问题 python3.5 + cuda9.0 + cudnn7.3.1

pip安装 pytorch问题

最近借了个服务器玩玩，基于linux16.04的服务器，但是服务器只安装了tensorflow，但我程序使用pytorch写的，要将程序重新写成tensorflow有点麻烦，于是我就准备在服务器上安装pytorch ，然后发现，有一个最省时的办法就是先直接安装torchvision,然后他就可以自动安装最新的pytorch了，版本是1.3.0

pip3 install torchvision

但当我安装好了之后，在调用的时候，发现pytorch中torch.cuda.is_available()返回的是False，这就表示CUDA不可用，没办法调用GPU跑程序，下面找找原因：

import torch
torch.cuda.is_available()

1. CUDA未安装的情况

如果你的机器还没有安装cuda，那么就可以在pytorch中输入：

这里的意思是，适合当前pytorch的CUDA版本是CUDA10.1 。然后直接安装对应的cuda版本就行了，安装的步骤网上有很多，这里就不写了。