前言
在服务器上创建虚拟环境,安装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的软链接。