如何在linux上安装GPU版本pytorch

前言

在服务器上创建虚拟环境，安装gpu版本torch或者tf框架。

一、查看服务器类型

查看linux的内核是centos还是ubuntu：

cat /etc/redhat-release 

二、虚拟环境中安装

1.查看自己的cuda版本

命令如下：

nvidia-smi

 右上角就是最高能安装的CUDA Version，我的是11.1

2.官网安装

去pytorch官网（PyTorch）进行查看安装命令：

 最好使用pip安装吧，试过conda会出现问题。安装cuda==11.0

 测试是否安装好：

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

如果之前虚拟环境利用cpu的pytorch，记得需要先删除。不然一直装不好。

3.CUDA多版本切换

如果安装了多个cuda版本后，可以在/usr/local/目录下查看自己安装的cuda版本，如下图所示：

或者使用如下命令： 

ls -l /usr/local | grep cud

 第一个是目前cuda的软链接。