简要记录遇到的问题。
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安装。
7、验证是否可以使用torch-gpu
import torch
torch.cuda.is_available()
返回TRUE则成功。
Original: https://blog.csdn.net/weixin_42423653/article/details/125412513
Author: 无内存靠心算
Title: Linux下安装CUDA及torch-gpu版
原创文章受到原创版权保护。转载请注明出处：https://www.johngo689.com/709496/
转载文章受原作者版权保护。转载请注明原作者出处！