1.安装NVIDIA驱动
1.1 禁用原有驱动
首先进入ubuntu系统,检查nouveau是否禁用,在终端输入
lsmod | grep nouveau
若有输出,则禁用原有的nouveau nvidia驱动,在终端输入
sudo gedit /etc/modprobe.d/blacklist.conf
在文件最后添加下面两行
blacklist nouveau
options nouveau modeset=0
更新系统修改
sudo update-initramfs -u
然后重启
sudo reboot
重启之后再次检查nouveau是否已经禁用,在终端输入下面命令,若无任何输出则禁用成功
lsmod | grep nouveau
1.2 完全卸载nvidia驱动
若之前有安装过nvidia驱动,则需要完全卸载,在终端输入以下命令
sudo apt-get --purge remove nvidia*
sudo apt-get --purge remove "*nvidia*"
sudo apt-get --purge remove "*cublas*" "cuda*"
sudo apt autoremove
sudo apt update
1.3 修改bios设置
开机按F2进入bios
将Security Boot这一项改为:Disabled
将Configuration这一项将图形设备Graghic Device选项改为:Switchable Graphics
1.4.1 安装NVIDIA驱动(方法一)
打开ubuntu系统,在软件和更新中先更换下载源,然后在附加驱动这一栏中选择一个驱动进行安装。安装完之后,在终端中输入
nvidia-smi
若没有报错,则安装成功;如果报错,那么使用下列方法进行
sudo dkms remove nvidia/显卡型号(数字,如:460.39) --all
sudo dkms install --force nvidia/显卡型号 -k $(uname -r)
sudo update-initramfs -u
sync
reboot
显卡型号使用该命令查看
ls /usr/src | grep nvidia
重启之后再次在终端输入
nvidia-smi
应该就成功了!!!
1.4.2 安装NVIDIA驱动(方法二)
参照链接: Ubuntu20.04安装NVIDIA显卡驱动+cuda+cudnn配置深度学习环境.
2.安装cuda
2.1 cuda安装
运行nvidia-msi时可以查看当前显卡最高支持的cuda版本,下载cuda的网址链接: https://developer.nvidia.com/cuda-toolkit-archive.
找到自己需要的cuda版本,然后最下面的虚线矿会给出下载和下载后的命令
在终端执行:
wget http://developer.download.nvidia.com/compute/cuda/11.0.2/local_installers/cuda_11.0.2_450.51.05_linux.run
但是下载到99%时,会报错“s段错误”,采用axel方法进行多线程下载
sudo apt install axel
#-n 50 指的是使用50个线程进行下载
axel -n 50 http://developer.download.nvidia.com/compute/cuda/11.0.2/local_installers/cuda_11.0.2_450.51.05_linux.run
下载完成后进行cuda的安装
cd到下载的cuda目录,然后执行命令
sudo sh cuda_11.0.2_450.51.05_linux.run
正常情况,会出现下面显示,在下方输入accept即可
在Drive处敲空格,选择不安装驱动,因为之前已经安装过驱动程序,然后光标选中最后的Install再回车,开始安装
显示如图所示,表示安装成功
2.2 配置环境变量
安装成功之后,配置环境变量,使用命令打开.bashrc文件
gedit ~/.bashrc
在文件末尾输入如下内容保存并退出
#将下面的11.0替换为你的cuda版本,其他不变,如果不知道自己安装的是哪个版本,就去/usr/local/文件夹下找一下
export PATH=/usr/local/cuda-11.0/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-11.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
保存退出之后,执行命令更新变量
source ~/.bashrc
然后在终端输入
nvcc -V
如果显示如图,则表示安装成功
3. 安装cudnn
下载cudnn的网址为
https://developer.nvidia.com/rdp/cudnn-download
找到对应版本的cuda版本进行下载。
解压下载好的cudnn压缩包,然后执行
# 将文件复制到cuda对应的文件夹下
sudo cp cuda/include/cudnn* /usr/local/cuda/include/
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/
#赋予文件执行权限
sudo chmod a+r /usr/local/cuda/include/cudnn*
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*
4. 测试是否安装成功
在终端输入
cd /usr/local/cuda/samples/1_Utilities/deviceQuery
sudo make
./deviceQuery
出现Result = Pass,说明安装成功