当前位置：首页 > news >正文

ubuntu, nvidia driver, cuda, cudnn， pytorch-gpu版本安装

news 2026/2/9 2:11:49

文章目录

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 驱动

https://blog.csdn.net/takedachia/article/details/130354110

3.ubuntu 安装 anaconda

https://developer.aliyun.com/article/988482
export PATH=“~/anaconda3/bin”:$PATH 导入变量 source ~/.bashrc使生效

conda create -n pytorch python=3.9

4.安装pytorch gpu版本

记住此时安装的cuda是不完整的，但是对于pytorch来说够用了。
https://pytorch.org/get-started/previous-versions/
可以看到：
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版本相关问题查看：
https://www.jb51.net/article/278075.htm
https://www.jianshu.com/p/eb5335708f2a
https://www.cnblogs.com/yhjoker/p/10972795.html

5.安装完整版cuda 和 cudnn

参考：
https://blog.csdn.net/takedachia/article/details/130375718
介绍的很详细。
上面的教程是通过https://docs.nvidia.com/deeplearning/cudnn/install-guide/index.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:也有多种安装方法
https://docs.nvidia.com/deeplearning/cudnn/install-guide/index.html