PyTorch安装教程（Linux）

安装准备

1. 请先安装Anaconda或者Miniconda并配置好清华源
2. 输入命令nvidia-smi，查看CUDA版本，可以看到是10.1
   

安装教程

1. 用conda新建一个名为pytroch的环境

conda create -n pytorch python=3.7

2. 激活刚才创建的名为pytorch的环境

conda activate pytorch

3. 进入Pytorch的官网，CUDA版本选择之前我们查询的。然后复制下方的安装命令
   
4. 在第2步激活的pytorch环境中运行第3步复制的安装命令

conda install pytorch torchvision cudatoolkit=10.1 -c pytorch

验证安装

1. 验证torch能不能正常使用，正常打印则说明安装成功

import torch
x = torch.rand(5, 3)
print(x)

2. 再来看看CUDA是否能用，如果打印True则说明能用

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

引用参考

https://pytorch.org/get-started/locally/