更新系统
首先运行下面的两行命令,对系统进行更新
sudo apt update
sudo apt upgrade
安装NVIDIA驱动
查找可用的驱动版本
运行下面的命令查找可用的驱动版本
ubuntu-drivers devices
安装选的驱动版本
运行下面的命令,安装我们选定的驱动版本
sudo apt install nvidia-driver-535
重启服务器
驱动安装完成之后一定要重启服务器,可以运行下面的命令重启服务器
reboot
如果你不是以root用户登录服务器的,则需要运行下面的命令来重启服务器
sudo reboot
查看驱动和显卡信息
重启之后执行 nvidia-smi,正常情况下可以看到 nvidia driver 信息以及显卡信息:
nvidia-smi
可以看到驱动的版本是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进行安装
官方地址:
下载与安装
先后运行下面的两行代码:
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
安装的时候记得去掉安装 driver 的选项,因为上面已经装过了。
安装完成之后,会出现如下的信息:
添加到环境变量
运行下面的三行命令,将上面安装的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的版本了!
原创文章,作者:朋远方,如若转载,请注明出处:https://caovan.com/ubuntu2204anzhuangnvidiaqudongcuda/.html