更新系统
首先运行下面的两行命令,对系统进行更新
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的版本了!
如果重新登录之后又出现如下的情况
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