700字范文,内容丰富有趣,生活中的好帮手!
700字范文 > 怎样查看cudnn版本_ubuntu16.04+Geforce GTX1060安装CUDA和cuDNN

怎样查看cudnn版本_ubuntu16.04+Geforce GTX1060安装CUDA和cuDNN

时间:2021-12-03 10:24:23

相关推荐

怎样查看cudnn版本_ubuntu16.04+Geforce GTX1060安装CUDA和cuDNN

DELL G7系列,ubuntu16.04,本机显卡型号: GeForce GTX 1060。

主要步骤如下:

一,安装NVIDIA驱动。

1,本笔记本是F12进入BIOS,将secure boot设置为disable。

2,通过下面命令确保电脑上之前的nvidia驱动已卸载干净(可选)。

sudo apt-get remove --purge nvidia*

3,通过下面命令卸载之前下载的驱动安装包(runfile格式)(可选)。

sudo sh /home/xxx/驱动名.run --uninstall

4,屏蔽开源驱动nouveau,这里选择blacklist,首先创建新文件。

sudo vim /etc/modprobe.d/blacklist-nouveau.conf

内容如下:

blacklist nouveauoptions nouveau modeset=0

再通过如下命令更新:

sudo update-initramfs -u

通过如下命令,nouveau驱动是否被禁用(无输出则被禁用)。

lsmod | grep nouveau

5,打开终端加入,加入官方ppa源。

sudo add-apt-repository ppa:graphics-drivers/ppa

6,刷新软件库。

sudo apt-get update

7,查看推荐的显卡驱动,选择驱动后边有推荐(recommended)字符的。

ubuntu-drivers devices

本机是推荐nvidia-430,建议不在NVIDIA官网下最新驱动。

8,通过下面命令安装推荐的显卡驱动,并在完成后重启。

sudo apt-get install nvidia-430 nvidia-settings nvidia-primereboot

9,检查(确保两个命令都有正常输出)。

下面命令显示显卡的一些内容信息。

nvidia-smi

设置显卡。

nvidia-settings

二,安装cuda和cudnn

1,cuda官网下载,/cuda-10.0-download-archive,选择runfile版本。

2,将run文件拷贝到HOME下面,在shell下,HOME目录里执行如下命令。将cuda_10.0.130_410.48_linux.run重命名为cuda.run。

sudo sh cuda.run

这里会显示用户认证信息,按回车键加载到100%。先accept,INSTALL NVIDIA ACCELERATED GRAPHICS DRIVER............选择no。

后面就一路yes和回车即可。

3,设置环境变量。

sudo gedit /etc/profile

并在文件末尾添加如下两行。

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}}

并执行如下命令,保证了环境变量的永久配置。

source /etc/profilereboot

4,验证驱动版本和CUDA版本。

cat /proc/driver/nvidia/version

nvcc -V

5,安装cudnn

cuDNN Archive​

需要先注册NVIDIA帐号,并通过如下命令解压(在下载好的目录里)。

tar -xvf cudnn-10.0-linux-x64-v7.3.1.20.tgz

解压下载的文件,可以看到cuda文件夹,在当前目录打开终端,执行如下命令。

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.hsudo chmod a+r /usr/local/cuda/lib64/libcudnn*

并查看cudnn版本。

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

参考链接:

Ubuntu16.04上CUDA_10.0及cuDNN的安装_lihe的博客-CSDN博客​

本内容不代表本网观点和政治立场,如有侵犯你的权益请联系我们处理。
网友评论
网友评论仅供其表达个人看法,并不表明网站立场。