ubuntu16.04安装cuda9.0和cudnn对应版本教程

    科技2022-08-10  94

    ubuntu16.04安装cuda9.0和cudnn对应版本教程

    为了学习过程中的环境需求,所以笔者需要搭建对应的学习环境。由于笔者接触ubuntu时间不长,所以在配置过程踩了很多的坑。在这里笔者把自己的经验给大家分享一下。 本次教程是在ubuntu16.04系统和拥有nvidia显卡驱动的基础上进行的。笔者驱动版本450.57,能够满足cuda9.0需要。

    1.CUDA安装

    1)CUDA9.0下载

    CUDA系列的包可以直接去NVIDIA官网进行下载,而且下载速度也比较快。读者可以根据自己电脑驱动版本去下载对应的版本。选择下载类型时,推荐使用runflie类型的文件,安装起来简单而且成功率较高。如图所示

    下载完毕后进入下载的目录终端,为了安装方便我将下载后的文件命名为cuda.run(不修改文件名也可直接输入原始名字),输入命令检测,序号不同需要重新下载。

    md5sum cuda.run

    2)安装cuda

    (1)禁用nouveau

    如果实现安装过驱动,那么应该禁用过nouoveay,可以使用命令进行查看:

    lsmod | grep nouveau

    如果没有输出,那么禁用成功,如果没有禁用过可在终端输入命令:

    sudo gedit /etc/modprobe.d/blacklist.conf

    然后会打开一个文档,再在文档的最后添加代码:

    blacklist nouveau options nouveau modeset=0

    主要目的是为了防止驱动发生冲突。 保存后退出,然后在终端输入命令,使修改的内容生效,等待重启:

    sudo update-initramfs -u reboot

    (2)在字符界面进行安装

    紧接上一步的重启电脑过后,可以直接使用Ctrl+Alt+F1进入字符界面,输入自己的帐号密码进行登录。登录成功后会出现终端一样的命令输入字符。关闭图形界面: sudo service lightdm stop 执行安装命令 sudo sh cuda.run

    会显示一大串的信息,主要是用户许可证的信息,按住空格键,最下方显示的进度会到100% 接着会出现进行选择的命令,输入内容为:

    accept -> n -> y -> y ->y (对于没有选项的命令输入回车)

    重启图形界面:

    sudo service lightdm start

    如果没有返回图形界面,使用快捷键 Ctrl+Alt+F7可返回图形界面。

    设置环境变量 打开终端输入命令: sudo gedit ~/.bashrc

    在打开的文档的最后方加入下列语句:

    export CUDA_HOME=/usr/local/cuda export PATH=$PATH:$CUDA_HOME/bin export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

    建议加上注释,方便以后添加和删除环境路径。

    (3)验证驱动版本和cuda版本

    cat /proc/driver/nvidia/version nvcc -V

    (4)编译例程

    编译例程文件: 打开终端输入命令: cd NVIDIA_CUDA-9.0_Samples make

    编译时间可能较长,请耐心等待。

    运行例程文件: 进入~/NVIDIA_CUDA-9.0_Samples/bin/x86_64/linux/release,可以使用命令进入也可通过文件夹进入在打开终端,输入命令: ./deviceQuery

    得到Result = PASS如图。

    检查系统和CUDA-Capable device的连接情况,终端输入 : ./bandwidthTest

    得到输出结果如下图所示,表示安装成功。

    3)安装cudnn

    (1)cudnn文件下载

    在NVIDIA官网下载:cudnn下载 选择最新版本进行下载,但是要注意匹配自己的cuda版本,选择第三个文件,cuDNN Library for linux。

    下载结束后会获得一.solitairetheme8的文件,我们需要将文件转换为.tgz类型的文件,方便提取。进入下载目录打终端输入命令进行

    cp cudnn-9.0-linux-x64-v7.6.3.30.solitairetheme8 cudnn-9.0-linux-x64-v7.6.3.30.tgz tar -xvf cudnn-9.0-linux-x64-v7.6.3.30.tgz

    然后在提取的目录下打开终端输入命令:

    sudo cp cuda/include/cudnn.h /usr/local/cuda-9.0/include/ sudo cp cuda/lib64/libcudnn* /usr/local/cuda-9.0/lib64/

    (2)查看cudnn版本

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

    此时我们就已经安装成功了。

    参考文献: [1]: https://blog.csdn.net/zbr794866300/article/details/106574704 [2]: https://blog.csdn.net/Super_jm_/article/details/78397074?utm_medium=distribute.pc_relevant.none-task-blog-BlogCommendFromBaidu-8.nonecase&depth_1-utm_source=distribute.pc_relevant.none-task-blog-BlogCommendFromBaidu-8.nonecase

    Processed: 0.050, SQL: 8