系统 Ubuntu 20.04
显卡 A5000
1. 安装驱动
1.1 清理已安装nvidia-driver和cuda-清理完后一定要重启
#清理 nvidia-driver
sudo apt-get remove --purge '^nvidia-.*'
sudo apt-get --purge remove "*nvidia*"
sudo apt-get install ubuntu-desktop
sudo rm /etc/X11/xorg.conf
echo 'nouveau' | sudo tee -a /etc/modules
sudo apt-get autoremove
sudo apt-get autoclean
sudo apt-get update
#清理 cuda
sudo apt-get --purge remove "*cublas*" "cuda*" "nsight*"
sudo rm -rf /usr/local/cuda*
sudo apt-get autoremove
sudo apt-get autoclean
sudo apt-get update
##重启电脑
sudo reboot
1.2 安装驱动,这里我们安装 470版本
sudo apt install nvidia-driver-470 nvidia-dkms-470
#安装完后重启
sudo reboot
2. 安装cuda
cuda我们选择11.4版本,实际上470显卡支持的最大版本也为11.4.
登陆网址CUDA Toolkit 11.4 Downloads | NVIDIA Developer, 照着下边依次选择。
运行代码,下载 并运行.run文件,开始安装。
#下载
wget https://developer.download.nvidia.com/compute/cuda/11.4.0/local_installers/cuda_11.4.0_470.42.01_linux.run
#运行
sudo sh cuda_11.4.0_470.42.01_linux.run
进入如下安装页面:
注意: 这里摁回车取消‘Driver’安装,否则会再次安装470显卡驱动。然后选择‘install’即可
安装完成以后将cuda添加进系统路径。终端输入代码进入.bashrc后
gedit ~/.basrch
在最下边添加以下两行
export PATH="/usr/local/cuda-11.4/bin:$PATH"
export LD_LIBRARY_PATH="/usr/local/cuda-11.4/lib64:$LD_LIBRARY_PATH"
3. 验证安装结果
#验证驱动安装
nvidia-smi
#验证cuda安装
nvcc -V