Linux下安装CUDA及torch-gpu版
linux下安装CUDA以及torch-gpu,root用户磁盘空间不够,于是转为在非root用户下安装,要更改环境变量。
简要记录遇到的问题。
1、输入nvidia-smi 不显示显卡信息(忘截图了)
直接重启服务器,输入
shutdown -r now
2、获取显卡信息后按照来链接网页指示下载nvidia
How to install Nvidia driver on CentOS 7 Linux - nixCraft (cyberciti.biz)
下载后运行
sudo bash NVIDIA-Linux-x86_64-515.48.07.run
3、下载安装cuda
一开始使用的是root账号,但是发现root空间不够,无法解压和安装,于是更改了tmp路径以及安装路径,想在root外指定的路径下安装,但是失败了(没有报错,但是一直没动静,可能是我自己机器的问题),当时输入的是 sudo sh cuda_11.7.0_515.43.04_linux.run --tmpdir=/data/tmp --installpath=/data/CUDA
Linux 下的 CUDA 安装和使用指南 - 知乎 (zhihu.com)
后转为在非root下安装CUDA,成功。
4、非root用户下安装cuda以及cudnn
(29条消息) Linux 非root用户安装CUDA,CUDNN_qq_41804016的博客-CSDN博客
安装过程基本一致,需要修改环境变量,中间用到了vim工具。
Vim使用教程(更改环境变量主要 使用了 vim ~/.bashrc (打开bashrc文件) ,插入(i)环境变量,按esc退出插入,然后输入 :wq 退出编辑,再source ~/.bashrc 保存。
Linux Vim基本操作(文件的打开和编辑)完全攻略(有图有真相) (biancheng.net)
5、验证是否安装成功
输入
nvcc -V
显示安装成功。
6、 安装torch的gpu版本(使用的是离线安装),自行下载相应的版本,然后 pip install ****.whl安装。
https://download.pytorch.org/whl/torch_stable.htmlhttps://download.pytorch.org/whl/torch_stable.html
7、验证是否可以使用torch-gpu
import torch
torch.cuda.is_available()
返回TRUE则成功。
更多推荐
所有评论(0)