NVIDIA显卡驱动及CUDA相关安装流程
NVIDIA驱动安装
NVIDIA驱动官方下载地址:
/Download/index.aspx?lang=cn
第一步先查看适合的显卡驱动版本
ubuntu-drivers devices
第二步下载合适的显卡驱动
第三步下载驱动相关依赖
安装lightdm是在弹窗上选择lightdm[若没有弹窗略过]
第四步禁用nouveau驱动
第五步禁用nouveau内核模块
第六步卸载旧的nvidia驱动【如果没装过,可以省略】
sudo apt-get --purge remove nvidia-*
第七步重启
sudo reboot
第八步禁用图形化界面
sudo service lightdm stop
第九步修改文件权限并安装
# 先进到NVIDIA驱动所在的文件夹# 赋予驱动权限sudo chmod a+x 驱动文件名# 安装sudo ./文件名 -no-x-check -no-nouveau-check -no-opengl-files
第十步开启图形化界面【若Xshell远程连接,不需要图形化界面也可略过】
service lightdm start
第十一步查看是否安装成功
nvidia-smi# 出现如下返回则安装成功
CUDA安装
cuda官方下载地址
/cuda-toolkit-archive
第一步下载相应版本
第二步放到服务器上运行安装
sudo sh 安装包名称
输入accept
去掉安装驱动选项,并选择Install【驱动上文已安装】
第三步配置cuda环境变量(以下三种皆可)
export PATH="/usr/local/cuda-11.0/bin:$PATH"export LD_LIBRARY_PATH="/usr/local/cuda-11.0/lib64:$LD_LIBRARY_PATH"
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
Cudnn安装
cudnn下载官网地址
/rdp/cudnn-download
第一步下载与cuda对应版本
第二步解压服务器上下载好的文件
tar -xvf 安装包名字.tar.xz
第三步进到解压的文件夹内部
# cp 后面指定自己下载的cuda版本
第四步至此已安装完成
Anaconda安装
Anaconda官方下载地址
/
第一步放到服务器上运行安装
sudo bash 安装包名称# 空格跳过文字,选择安装路径# [/root/anaconda3] >>> /usr/anaconda3
第二步修改环境变量
第三步激活虚拟环境
多版本CUDA切换
第一步下载与cuda版本适配的gcc与g++
# 例如cuda10需要gcc4.8与g++4.8sudo apt-get install gcc-4.8 g++-4.8# 查询已安装版本ls /usr/bin/gcc*
第二步进行gcc与g++的版本切换
# gccsudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-4.8 40sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-9 90# g++sudo update-alternatives --install /usr/bin/g++ g++ /usr/bin/g++-4.8 40sudo update-alternatives --install /usr/bin/g++ g++ /usr/bin/g++-9 90# 数字40,90代表优先级,数字越大优先级越高# 若要移除(否则不用执行)sudo update-alternatives --remove gcc /usr/bin/gcc-4.8# 输入切换命令(切换g++只需要将gcc替换成g++即可)sudo update-alternatives --config gcc# 输入需要选择的版本对应序号(0,1,2)
第三步安装cuda(步骤与上面一样)
# 安装完修改软连接# sudo vim ~/.bashrcexport LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64export PATH=$PATH:/usr/local/cuda/binexport CUDA_HOME=$CUDA_HOME:/usr/local/cuda# 更新环境变量# source ~/.bashrc
第四步安装cudnn(步骤与上面一样)
第五步切换cuda
# 删除默认链接sudo rm -rf /usr/local/cuda# 指定版本链接sudo ln -s /usr/local/cuda-10.0 /usr/local/cuda
第六步查看版本是否切换成功
# 查看cuda版本nvcc -V# 查看cudnn版本cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
on
删除默认链接
sudo rm -rf /usr/local/cuda
指定版本链接
sudo ln -s /usr/local/cuda-10.0 /usr/local/cuda
第六步查看版本是否切换成功~~~python# 查看cuda版本nvcc -V# 查看cudnn版本cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2