Pytorch安装教程（Linux)

请先安装Anaconda或者Miniconda并配置好清华源

输入命令nvidia-smi，查看CUDA版本，可以看到是10.1

安装教程

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

conda create -n pytorch python=3.7

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

conda activate pytorch

进入Pytorch的官网，CUDA版本选择之前我们查询的。然后复制下方的安装命令

在第2步激活的pytorch环境中运行第3步复制的安装命令

conda install pytorch torchvision cudatoolkit=10.1 -c pytorch 验证安装

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

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

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

import torch print(torch.cuda.is_available()) 引用参考 https://pytorch.org/get-started/locally/