基于conda安装GPU版本的pytorch

安装cuda

从官网上下载或者从清华镜像下载都可以；
安装对应Ubuntu系统和nvidia驱动匹配的cuda系统：

安装cuda

conda install cudatoolkit == 10.1

安装Cudnn

安装cudnn：

conda install cudnn

激活cudnn：
打开.bashrc文件，在文件内部添加：

export PATH="/usr/local/cuda-10.1/bin:$PATH"
export LD_LIBRARY_PATH="/usr/local/cuda-10.1/lib64:$LD_LIBRARY_PATH"

再激活：

source ~/.bashrc

安装pytorch

一定要找到对应的版本进行安装，不能直接使用指令：

conda install torch

做法：
去torch的官网上找到和cuda版本对应的torch下载指令，进行下载；
如果没有的话可以去previous版本里面进行查找；
测试：
如果需要使用GPU的需要测试能否在GPU上使用：

python
import torch
torch.cuda.is_available()

如果结果是True就表示可以在GPU上使用。
切记，一定要按照上述的方法，对应好所有软件的版本号，否则torch.cuda.is_available()都有False的风险。