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

首先安装linux对应版本的Anaconda环境

这里我下载的是

Anaconda3-5.2.0-Linux-x86_64.sh

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

create -n pytorch python=3.8

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

conda activate pytorch

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

conda install pytorch1.8.0 torchvision0.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())

转载于作者：杭漂一族小张 https://www.bilibili.com/read/cv15156700/ 出处：bilibili