安装显卡驱动以及CUDA、cudnn
程序员文章站
2024-03-19 18:13:40
...
安装显卡驱动以及CUDA、cudnn
ubantu安装显卡驱动以及CUDA、cudnn
安装显卡驱动
参考链接
1、查看显卡型号
lshw -c video
结果如下:我的型号是GF119M [GeForce 610M]
2、查找显卡驱动
去官网查看显卡驱动
下面是我的电脑对应的驱动版本:
搜索的结果页面看到,我的驱动版本应该是390.132
为了再次确认一遍,你还可以使用这个命令查看你可以使用的驱动:
ubuntu-drivers devices
结果显示
如果结果显示和搜索到的驱动版本一样,推荐也是390.132,那我们可以安心安装390.132版本。但是结果显示的并没有390.132,为了避免出错,建议选择结果显示中的版本号384。
3、安装显卡驱动
1、禁用nouveau
sudo gedit /etc/modprobe.d/blacklist.conf
在末尾加上下面两句话
blacklist nouveau
options nouveau modeset=0
2、执行命令:
sudo update-initramfs -u //应用更改
3、重启系统,然后输入验证是否禁用
lsmod | grep nouveau
显示如图则说明成功(图为别人的图,自己忘记截取了):
4、安装显卡驱动
打开系统设置->软件和更新->附加驱动->选择NVIDIA驱动->应用更改(如图)
4、查看NVIDIA版本检验是否安装完成
nvidia-smi
安装CUDA
安装cuda注意与cudnn版本的匹配本文安装的是cuda9
1、下载安装文件
下载链接
2、安装cuda
进入下载安装文件的文件夹,然后运行以下命令:
sudo sh cuda_9.0.176_384.81_linux.run
然后按回车键或者空格键知道条款显示100%,然后如下图操作:
3、设置环境变量
打开主目录下的 .bashrc文件添加路径
sudo gedit ~/.bashrc
在末尾添加:
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-9.0/lib64
export PATH=$PATH:/usr/local/cuda-9.0/bin
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda-9.0
然后运行
source ~/.bashrc
4、检查是否安装成功
nvcc --version
如果显示如下,则安装成功
安装cudnn
1、下载安装文件
官网下载安装文件我下载的是7.6.4版本
注意选择第三个下载(下载需要注册)
2、安装文件
下载完成后解压
然后执行如下命令
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*
3、检查
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
如下则安装成功
推荐阅读
-
Windows/Ubuntu下查看显卡,驱动,cuda,cudnn版本
-
安装显卡驱动以及CUDA、cudnn
-
centos7.6 安装NVIDIArtc2070显卡驱动
-
Ubuntu 16.04 安装NVIDIA显卡驱动及CUDA、CuDnn(详细)
-
ubuntu16.04安装Nvidia显卡驱动、CUDA、cudnn
-
Ubuntu 16.04 安装Nvidia 驱动及cuda10.2
-
ubuntu16+NVIDIA驱动+CUDA+cuDNN配置安装
-
Ubuntu安装nvidia显卡驱动和CUDA CUDNN autoinstall
-
ubuntu18.04:安装nvidia-440驱动+cuda10.2+cudnn8.0+pytorch
-
基于Ubuntu20.04安装Nvidia驱动、Cuda和CudNN