ubuntu20.04-cuda11.0-cudnn8.0.5 安装
卸载显卡驱动
sudo nvidia-uninstall
查看一下
sudo find -iname nvidia
删除相关文件
sudo apt-get purge nvidia-*
sudo apt-get remove --purge nvidia-\*
sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt-get update
安装显卡驱动
sudo apt-get install nvidia-390
根据需要修改 nvidia-390,显卡安装完,重启机器.
CUDA
sudo dpkg -i cuda-repo-ubuntu1710-9-2-local_9.2.148-1_amd64.deb
sudo apt-key add /var/cuda-repo-<version>/7fa2af80.pub
sudo apt-get update
sudo apt-get install cuda
sudo chmod +x cuda_11.0.3_450.51.06_linux.run
sudo ./cuda_11.0.3_450.51.06_linux.run -no-x-check -no-nouveau-check -no-opengl-files
需要把Driver给去掉,否则会报错,因为已经装了Driver
可查看官网:
CUDA download
/etc/profile 添加以下内容
sudo vim /etc/profile
或者
sudo vim /etc/bash.bashrc
export CUDA_HOME=/usr/local/cuda
export PATH=$PATH:$CUDA_HOME/bin
export LD_LIBRARY_PATH=$CUDA_HOME/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
测试cuda是否安装成功
cd /usr/local/cuda/samples/1_Utilities/deviceQuery
sudo make
./deviceQuery
卸载
cd /usr/local/cuda-11.0/bin/
sudo ./cuda-uninstaller
sudo rm -rf /usr/local/cuda-11.0
其他
查看显卡信息
扫描二维码关注公众号,回复:
16253884 查看本文章
lspci|grep NVIDIA
CUDNN
配置安装
tar -zxf cudnn-11.0-linux-x64-v8.0.5.39.tgz
sudo cp cuda/include/* /usr/local/cuda/include/
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*
cuda 和cudnn版本
查看官网
cuda-toolkit
查看 cuda 版本
cat /usr/local/cuda/version.txt
查看 cudnn 版本
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
备注
如果报错,请相信我,直接重新安装系统,或者节省的时间更多。