前言

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

Logo

华为云1024程序员节送福利,参与活动赢单人4000元礼包,更有热门技术干货免费学习

更多推荐