Ubuntu16安装cuda和cudnn的一点问题及解决方法
ubuntu 16.04默认安装了第三方开源的驱动程序nouveau,安装nvidia显卡驱动首先需要禁用nouveau,不然会碰到冲突的问题,导致无法安装nvidia显卡驱动。
sudo gedit /etc/modprobe.d/blacklist.conf ->打开此文件。
blacklist nouveau->在文件结尾输入,保存并退出。
sudo update-initramfs -u ->执行,让修改后的文件生效。
重启电脑
lsmod | grep nouveau ->测试nouveau是否禁用成功,是的话,执行命令后应该什么都没出现。
https://blog.csdn.net/qq_42730894/article/details/107655026 ->之后的操作。
安装cuda10.1:
下载驱动支持对应的.run文件。
放到home文件夹下。
sudo sh xxx.run ->执行
按提示,添加环境变量。sudo gedit /etc/profile
在文件最后添加:
export PATH=/usr/local/cuda-10.1/binKaTeX parse error: Expected '}', got 'EOF' at end of input: {PATH:+:{PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-10.1/lib64KaTeX parse error: Expected '}', got 'EOF' at end of input: …LIBRARY_PATH:+:{LD_LIBRARY_PATH}}
重启电脑。
nvcc -V验证cuda是否安装成功。
安装cudnn 7.6.3 for cuda10.1
下载cudnn for linux.tar。
解压出一个cuda文件夹(tar -xzvf cudnn-9.0-linux-x64-v7.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 /usr/local/cuda/lib64/libcudnn*
测试cudnn是否可用:
from torch.backends import cudnn
print(cudnn.is_available())