如何在linux服务器上安装pytorch-gpu

现在一般租的远程服务器都会自带Anaconda环境

我们只要创建对应的虚拟环境，简单配制即可

这里我的电脑配置是2080ti服务器

create -n pytorch python=3.8

创建了一个python3.8的虚拟环境，再进入虚拟环境

conda activate pytorch

然后在pytorch官网找到安装命令，

conda install pytorch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0 cudatoolkit=10.2 -c pytorch

下载了pytorch1.8版本，cuda10.2，下载速度还是很快的

安装完成之后测试一下

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

安装完成