欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

ubuntu 18.04 笔记本双显卡 配置 cuda 与 cudnn

程序员文章站 2022-06-01 13:17:47
...

英伟达官网下载对应驱动http://www.nvidia.cn/Download/index.aspx

1.卸载现有驱动

sudo apt-get remove --purge nvidia*

2.禁用 显卡驱动

 

sudo gedit  /etc/modprobe.d/blacklist-nouveau.conf
blacklist nouveau
blacklist lbm-nouveau
options nouveau modeset=0
alias nouveau off
alias lbm-nouveau off

保存

echo options nouveau modeset=0 | sudo tee -a /etc/modprobe.d/nouveau-kms.conf

重启

sudo update-initramfs -u

sudo reboot
 sudo apt-get install linux-source
apt-get install linux-headers-x.x.x-genneric
sudo apt-get install linux-headers-$(uname -r)

安装 cuda10

sudo sh cuda__linux.run

modprobe nvidia

检查 nvidia-smi

export PATH=/usr/local/cuda-10.0/bin ${PATH:+: ${PATH}}

export LD_LIBRARY_PATH=/usr/local/cuda-10.0/lib64\
{LD_LIBRARY_PATH:+: ${LD_LIBRARY_PATH}}

复制

sudo cp cuda/include/cudnn.h /usr/local/cuda/include/
 
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/
 
sudo chmod a+r /usr/local/cuda/include/cudnn.h
 
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*