Linux下安装CUDA及torch-gpu版

简要记录遇到的问题。

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则成功。