cuda driver install
准备阶段
首先确认 cuda xserver 版本,保证和自己电脑显卡、cuda toolkit 版本适配,相关信息:
https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html
然后下载 driver 文件:
https://www.nvidia.cn/geforce/drivers/
卸载之前版本 nvidia driver :
sudo apt-get remove --purge nvidia*
一般来说,默认 driver 是 nouveau (对于ubuntu), 所以它需要被禁用 (注意修改版本对应路径):
cd /lib/modules/x.x.x-y-generic/kernel/drivers/gpu/drm/nouveau
mv nouveau.ko nouveau.ko.org
update-initramfs -u
或者
sudo vim /etc/modprobe.d/blacklist.conf
然后把这两行加在文件最后:
blacklist nouveau
options nouveau modeset=0
// 保存后执行
update-initramfs -u
重新生成 kernel initramfs 并且重启,最后检查:
lsmod | grep nouveau
没显示信息的话就拿捏了
安装
开始安装:
sudo chmod a+x NVIDIA-Linux-x86_64-xxx.xx.run
sudo ./NVIDIA-Linux-x86_64-xxx.xx.run -no-opengl-files
最后检查显卡能不能被正常识别,在设置里面看,或者用命令:
nvidia-smi
cuda toolkit install
到官网,用对应版本的命令:
https://developer.nvidia.com/cuda-toolkit-archive
在下载之前先复制 key ,不然他会报错,这里我用官网步骤就是这样,按报错信息稍微变换一下命令执行顺序就行。
然后更新配置文件:
sudo vim ~/.bashrc
把这两行加到最后:
export PATH=/usr/local/cuda-x.x/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
最后,再次更新配置文件并且检查其是否正常安装:
source ~/.bashrc
nvcc --version
如果找不到 nvcc 的话, 在 bashrc 文件加上这条命令,再更新一次:
export PATH=$PATH:/usr/local/cuda/bin
确保这个目录是可以找到 nvcc 可执行文件的。
cudnn install
到官网下载:
https://developer.nvidia.com/rdp/cudnn-download
按照 installation guide 进行, 如果在用 mnistCUDNN 测试时有这种报错:
fatal error: FreeImage.h: No such file or directory
那就安装依赖包:
sudo apt-get install libfreeimage3 libfreeimage-dev
之后理论上就都OK了
标签:nouveau,sudo,nvidia,PATH,cuda,显卡,驱动,NVIDIA From: https://www.cnblogs.com/gao0420/p/17927776.html