在Linux远程服务器上设置PyTorch的GPU版本

一般租远程服务器的话，都是租 linux 系统的

本文就以在 2080ti 显卡服务器上安装 pytorch-gpu 版本为例。

首先输入 python

可以看到这个服务器是已经安装好 anaconda 了，我们直接使用 conda 命令进行安装即可

conda info -e

可以看到我之前安装的 tensorflow1.13 版本 gpu 环境，想直接怎么安装 tf1.13.1 版本 gpu 的

可以看我之前写的博客，传送门

首先创建一个虚拟环境，我这里安装 pytorch1.7.1 版本的，python 为 3.7 版本，大家也可以

根据自己需求，安装相应的版本，命名也可以随意，比如 pytorch1.7.1 这个换成 py1.7 是不

conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=10.1 -c pytorch

最后输入 python

import torch

print(torch.cuda.is_available())