Ubuntu安装CUDA和cuDNN

Ubuntu安装cuda和cudnn
1、禁用nouvea
sudo gedit /etc/modprobe.d/blacklist.conf
在文件末端加入
blacklist nouveau
options nouveau modeset=0
然后更新
sudo update-initramfs -u
重启sudo reboot
输入 lsmod | grep nouveau查看是否禁用成功
如果禁用失败,查看在conf文件中的命令是否写错,或者删除nouveau,实在不行重装系统
2、安装英伟达显卡驱动
sudo ubuntu-drivers devices
建议装带recommend的
上官网搜对应驱动,下载
然后启动.run文件,启动,run文件网上找命令
nvidia-smi查看驱动是否安装成功
3、安装cuda
在英伟达官网寻找nvidia-smi给出的信息所匹配的cuda
按照官网给的命令安装,运行run文件后
跳转continue
一路enter
安装路径:
cd ~
sudo gedit .bashrc
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:/usr/local/cuda/extras/CPUTI/lib64
export CUDA_HOME=/usr/local/cuda/bin
export PATH= P A T H : PATH: PATH:LD_LIBRARY_PATH:$CUDA_HOME
4、安装cudnn
官网下载cudnn后,解压
tar -zxvf cudnn-9.0-linux-x64-v7.1.tgz
文件夹复制粘贴
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*
ubuntu普通用户使用cuda
直接将管理员的映射过来就行

Ubuntu安装CUDA和cuDNN_第1张图片
![在这里插入图片描述](https://img-blog.csdnimg.cn/e9a在这里插入图片描述
在vim编辑器最末尾中插入第二张图的命令
a编辑,shift+zz保存退出vim编辑器

你可能感兴趣的:(常用库和驱动安装,ubuntu,深度学习,pytorch)