Ubuntu22.04安装NVIDIA驱动+cuda

更新系统

首先运行下面的两行命令,对系统进行更新

sudo apt update
sudo apt upgrade

Ubuntu22.04安装NVIDIA驱动+cuda

安装NVIDIA驱动

查找可用的驱动版本

运行下面的命令查找可用的驱动版本

ubuntu-drivers devices

Ubuntu22.04安装NVIDIA驱动+cuda

安装选的驱动版本

运行下面的命令,安装我们选定的驱动版本

sudo apt install nvidia-driver-535

Ubuntu22.04安装NVIDIA驱动+cuda

重启服务器

驱动安装完成之后一定要重启服务器,可以运行下面的命令重启服务器

reboot

如果你不是以root用户登录服务器的,则需要运行下面的命令来重启服务器

sudo reboot

Ubuntu22.04安装NVIDIA驱动+cuda

查看驱动和显卡信息

重启之后执行 nvidia-smi,正常情况下可以看到 nvidia driver 信息以及显卡信息:

nvidia-smi

Ubuntu22.04安装NVIDIA驱动+cuda

可以看到驱动的版本是535.183.01,而后面的“CUDA Version:12.2”则代表可以支持cuda版本的最大版本号(并非实际的cuda版本)!

卸载驱动的方法(备忘,非必选步骤!)

sudo apt-get purge nvidia-*
sudo apt-get update
sudo apt-get autoremove

安装CUDA

案例中选择11.8版的cuda进行安装

官方地址:

https://developer.nvidia.com/cuda-11-8-0-download-archive?target_os=Linux&target_arch=x86_64&Distribution=Ubuntu&target_version=22.04&target_type=runfile_local

Ubuntu22.04安装NVIDIA驱动+cuda

下载与安装

先后运行下面的两行代码:

wget https://developer.download.nvidia.com/compute/cuda/11.8.0/local_installers/cuda_11.8.0_520.61.05_linux.run
sudo sh cuda_11.8.0_520.61.05_linux.run

Ubuntu22.04安装NVIDIA驱动+cuda

Ubuntu22.04安装NVIDIA驱动+cuda

安装的时候记得去掉安装 driver 的选项,因为上面已经装过了。

安装完成之后,会出现如下的信息:

Ubuntu22.04安装NVIDIA驱动+cuda

添加到环境变量

运行下面的三行命令,将上面安装的cuda添加到环境变量,cuda的版本要根据你自己的实际情况进行修改:

export PATH=$PATH:/usr/local/cuda-11.8/bin
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-11.8/lib64
export LIBRARY_PATH=$LIBRARY_PATH:/usr/local/cuda-11.8/lib64

添加到环境变量之后,你就可以通过“nvcc –version”命令来查看cuda的版本了!

Ubuntu22.04安装NVIDIA驱动+cuda

如果重新登录之后又出现如下的情况

caovan@caovan-Super-Server:~$ nvcc --version
找不到命令 “nvcc”,但可以通过以下软件包安装它:
sudo apt install nvidia-cuda-toolkit

可以尝试将这些 export 命令添加到你的 ~/.bashrc~/.bash_profile 文件中。这样,每次登录时都会自动加载这些设置。你可以用以下命令来编辑文件:

sudo vim ~/.bashrc

然后在文件末尾添加你的 export 命令,保存并退出。最后运行以下命令使更改生效:

source ~/.bashrc

原创文章,作者:朋远方,如若转载,请注明出处:https://caovan.com/ubuntu2204anzhuangnvidiaqudongcuda/.html

Like (0)
Donate 微信扫一扫 微信扫一扫
朋远方的头像朋远方
Previous 2024年10月10日 上午9:02
Next 2024年10月12日 下午3:00

相关推荐

发表回复

Please Login to Comment