显卡驱动安装
nvidia 官网找到显卡适配的驱动: /Download/index.aspx Installer Type 选择 runfile (local) 禁用 nouveausudo gedit /etc/modprobe.d/blacklist.conf # 在blacklist.conf文件末尾加上 blacklist nouveausudo update-initramfs -u # 更新reboot lsmod | grep nouveau # 没有输出说明禁用成功
安装依赖
sudo apt-get updatesudo apt-get install gccsudo apt-get install ubuntu-makesudo apt-get install make
安装驱动
sudo apt-get remove –purge nvidia* # 删除旧版本驱动sudo chmod a+x NVIDIA-Linux-x86_64-515.57.run # 添加可执行权限sudo ./NVIDIA-Linux-x86_64-515.57.run -no-x-check -no-nouveau-check -no-opengl-files # 以下两个选择与默认相反,其他选择与默认一致# 1. Install NVIDIA’s 32-bit compatibility libraries 选择与默认相反的选项: no# 2. Would you like to run the nvidia-xconfigutility to automatically update your x configuration so that the NVIDIA x driver will be used when you restart x? Any pre-existing x confile will be backed up. 选择与默认相反选项: yes
检验是否安装成功
nvidia-smi # 有显存占用率等信息的表格出现代表安装成功
CUDA 11.7 安装
nvidia 官网找到 CUDA 安装包: /cuda-downloads
Installer Type 选择runfile (local)
安装依赖
sudo apt-get install build-essential
下载安装包并安装
wget https://developer./compute/cuda/11.7.0/local_installers/cuda_11.7.0_515.43.04_linux.runsudo sh cuda_11.7.0_515.43.04_linux.run
cuDNN 8.4.1.50 安装
前往 cuDNN 官网下载源文件: /rdp/cudnn-download解压并复制 cuDNN 库文件到 CUDA 的库文件目录下(注意:这里也可以单独给 cuDNN 设置到一个单独目录下并添加环境变量,更利于多版本管理)
mkdir cudnn_8.4.1.50tar xvf cudnn-linux-x86_64-8.4.1.50_cuda11.6-archive.tar.xz -C cudnn_8.4.1.50cd cudnn_8.4.1.50sudo cp cuda/include/cudnn.h /usr/local/cuda/include # 注意看下 cudnn.h 目录结构是否正确sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64sudo chmod a+r /usr/local/cuda/include/cudnn.h sudo chmod a+r /usr/local/cuda/lib64/libcudnn*
多 CUDA 版本共存 (CUDA11.3 安装)
官网下载安装包: /cuda-11.3.0-download-archive?target_os=Linux参考以上 CUDA 11.7 的安装方式进行安装修改/usr/local/cuda
到需要的版本即可