ubuntu, nvidia driver, cuda, cudnn， pytorch

driver, cuda, cudnn， pytorch"/>

ubuntu, nvidia driver, cuda, cudnn， pytorch

文章目录

• 1.常用指令
• • • 1.1查看cpu是intel还是amd:
    • 1.2.查看ubuntu版本
    • 1.3.查看架构
    • 1.4.查看已安装的nvidia驱动
    • 1.5.进入tty模式
• 2.安装ubuntu22.04 和 nvidia 驱动
• 3.ubuntu 安装 anaconda
• 4.安装pytorch gpu版本
• 5.安装完整版cuda 和 cudnn
• 6.nvidia-driver, cuda-toolkit, cudnn

1.常用指令

1.1查看cpu是intel还是amd:

cat /proc/cpuinfo
或者lscpu

1.2.查看ubuntu版本

lsb_release -a

1.3.查看架构

arch
或者
uname -a

1.4.查看已安装的nvidia驱动

nvidia-detector

1.5.进入tty模式

Ctrl + Alt + F3进入tty模式
Ctrl + Alt + Delete 退出tty模式

2.安装ubuntu22.04 和 nvidia 驱动

3.ubuntu 安装 anaconda

export PATH=“~/anaconda3/bin”:$PATH 导入变量 source ~/.bashrc使生效

conda create -n pytorch python=3.9

4.安装pytorch gpu版本

记住此时安装的cuda是不完整的，但是对于pytorch来说够用了。
/
可以看到：
conda install pytorch1.13.1 torchvision0.14.1 torchaudio==0.13.1 pytorch-cuda=11.7 -c pytorch -c nvidia
这样安装以后gpu版本的pytorch就可以使用了。

但是此时 /usr/local/ 下面并没有cuda.

关于 nvidia-smi 和 nvcc --version 以及 print(torch.version.cuda) 不一致等cuda版本相关问题查看：
.htm

.html

5.安装完整版cuda 和 cudnn

参考：

介绍的很详细。
上面的教程是通过.html Package Manager Installation的方法安装cudnn的

安装后的 头文件和so文件 在

也可以利用tar file installation方法：
不需要安装，只需要解压和权限设置即可

6.nvidia-driver, cuda-toolkit, cudnn

除了以上的 通过下runfile来安装，也可以通过包安装
比如安装nvidia driver:
ubuntu-drivers devices 寻找可以安装的驱动

sudo ubuntu-drivers autoinstall 安装推荐
或者
sudo apt-get install cuda-drivers-418 安装特定版本的驱动。
nvidia-smi确认驱动安装正确

安装cuda:
conda search -c conda-forge cudatoolkit # 列出所有可以安装的版本 conda install -c conda-forge cudatoolkit=xx.x.x

nvcc-V 确定安装成功

cudnn:也有多种安装方法
.html