Ubuntu安装Nvidia 显卡驱动、CUDA Tookit、CuDNN极简指南(不想浪费生命了)

有一块显卡,怎么在Ubuntu中安装驱动和cuda那些依赖?

要快速安装。

Ubuntu安装Nvidia 显卡驱动、CUDA Tookit、CuDNN极简指南(不想浪费生命了)

如果之前安装了,就卸载:

### 卸载
sudo apt remove --purge nvidia-*
卸载依赖
sudo apt autoremove --purge

版本如何对应

https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html#id3

驱动版本最低的适配版本

Ubuntu安装Nvidia 显卡驱动、CUDA Tookit、CuDNN极简指南(不想浪费生命了)

不想细看,就使用下面命令行输入:

Nvidia Driver

sudo apt-get nvidia-common
sudo add-apt-repository ppa:graphics-drivers
sudo apt update
ubuntu-drivers devices
sudo apt install nvidia-driver-530

ubuntu-drivers devices:

Ubuntu安装Nvidia 显卡驱动、CUDA Tookit、CuDNN极简指南(不想浪费生命了)

cuda-toolkit

版本:

https://developer.nvidia.com/cuda-toolkit-archive

直接选最新的:

https://developer.nvidia.com/cuda-downloads

Ubuntu安装Nvidia 显卡驱动、CUDA Tookit、CuDNN极简指南(不想浪费生命了)

本地安装:

就是下载到本地后安装,本地会有一个deb文件

根据上图选择好后,下面会有一串操作命令,挨个输入就可以:

wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/cuda-ubuntu2204.pin

sudo mv cuda-ubuntu2204.pin /etc/apt/preferences.d/cuda-repository-pin-600

wget https://developer.download.nvidia.com/compute/cuda/12.1.1/local_installers/cuda-repo-ubuntu2204-12-1-local_12.1.1-530.30.02-1_amd64.deb

sudo dpkg -i cuda-repo-ubuntu2204-12-1-local_12.1.1-530.30.02-1_amd64.deb

sudo cp /var/cuda-repo-ubuntu2204-12-1-local/cuda-*-keyring.gpg /usr/share/keyrings/

sudo apt-get update

sudo apt-get -y install cuda

搞完重启: sudo reboot

安装CuDNN

Ubuntu安装Nvidia 显卡驱动、CUDA Tookit、CuDNN极简指南(不想浪费生命了)

https://developer.nvidia.com/rdp/cudnn-download

下载下来压缩包

Ubuntu安装Nvidia 显卡驱动、CUDA Tookit、CuDNN极简指南(不想浪费生命了)

点击查看下载哪个版本更合适:

Ubuntu安装Nvidia 显卡驱动、CUDA Tookit、CuDNN极简指南(不想浪费生命了)

如何查看目前系统和硬件信息?

查看Ubuntu信息

cat /proc/version

显示linux的内核版本和系统是多少位的:X86_64代表系统是64位的:

uname -a

要查看分区情况,那么输入命令:sudo fdisk -l

free -m 命令,这样就可以在命令字符界面查看内存的情况

解压: tar -xvf 文件.tar.gz

然后拷贝:

cd 文件夹名
sudo cp include/* /usr/local/cuda-12.1/include
sudo cp lib/libcudnn* /usr/local/cuda-12.1/lib64
sudo chmod a+r /usr/local/cuda-12.1/include/cudnn*
sudo chmod a+r /usr/local/cuda-12.1/lib64/libcudnn*

最后
cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2

显示:

Ubuntu安装Nvidia 显卡驱动、CUDA Tookit、CuDNN极简指南(不想浪费生命了)

配置环境

sudo vim ~/.bashrc

在最后添加:

export PATH=/usr/local/cuda-12.1/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-12.1/lib64

生效重载:

source ~/.bashrc

sudo ldconfig

验证

验证显卡驱动版本

cat /proc/driver/nvidia/version

查看:

nvcc -V

Ubuntu安装Nvidia 显卡驱动、CUDA Tookit、CuDNN极简指南(不想浪费生命了)

nvidia-smi

Ubuntu安装Nvidia 显卡驱动、CUDA Tookit、CuDNN极简指南(不想浪费生命了)

其他

如果项目需要的cuda、Torch版本等比较低怎么办?

每个环境可以单独安装cuda版本,上面安装的是系统的驱动和依赖包

我们可以使用conda或者pip安装。

conda和pip 环境内安装卸载

conda安装:conda install cuda -c nvidia

conda卸载:conda remove cuda

安装版本:conda install cuda -c nvidia/label/cuda-11.3.0

pip安装:

https://docs.nvidia.com/cuda/cuda-installation-guide-linux/index.html#pip-wheels

python3 -m pip install --upgrade setuptools pip wheel

python3 -m pip install nvidia-pyindex

|requirements.txt file: --extra-index-url <https://pypi.org/simple>

python3 -m pip install nvidia-cuda-runtime-cu12

我把12.1相关的驱动和安装包放在网盘了,需要自取

https://pan.baidu.com/s/1jM-JtLA3vtjB_1zDvUy3GA?pwd=9i4n

最后欢迎关注公众号:

Ubuntu安装Nvidia 显卡驱动、CUDA Tookit、CuDNN极简指南(不想浪费生命了)

原创文章。转载请注明: 作者:JiangYuan 网址: https://www.icnma.com
Like (0)
JiangYuan管理
Previous 23/05/2023 16:49
Next 27/05/2023 15:57

猜你想看

Leave a Reply

Please Login to Comment