美文网首页程序员
Ubuntu16系统显卡安装以及CUDA,CUDNN的安装(Ge

Ubuntu16系统显卡安装以及CUDA,CUDNN的安装(Ge

作者: 竹马不是马 | 来源:发表于2019-10-21 20:14 被阅读0次

    1、显卡驱动安装

    我在这里选用的ubuntu16系统,因为ubuntu16在ubuntu中算是经典系统,用户基数也大,各种环境基本上也能适配。系统在安装好系统后,进行系统软件源更新

    sudo apt update

    更新好系统源之后,在系统设置->软件和更新->附加驱动里选择NVIDIA驱动,因为我的显卡是GTX1060,所以在这里他给我推荐的是NVIDIA384驱动,点击应用之后等待它慢慢安装完成之后重启即可。

    重启之后打开终端输入nvidia-smi查看显卡驱动是否安装成功

    其他方法参考下面的博客,个人推荐第二种方法,博主讲得十分详细:

    https://blog.csdn.net/u014682691/article/details/80605201

    因为可能部分电脑安装完ubuntu系统后,系统不会检测出需要安装的驱动程序,所以上面链接中第二种方法是最为简单实用的。

    2、CUDA安装

    NVIDIA-linux-384对应的cuda版本是9.0,具体对应关系如下图

    Linux系统下对应关系

    进入英伟达cuda专区下载对应版本cuda,我在这里下载的是cuda9.0,文件类型最好是要选择run file,基本不会出错

    然后将cuda文件拷贝到HOME文件夹,按住control+alt+F1进入命令行界面,登陆上自己账号

    输入命令sudo sh /home/username/cuda_9.0.176_384.81_linux.run进行安装

    具体安装过程按照指引来进行就行,需要注意的是在安装cuda时系统也会默认给你安装显卡驱动,在这里我们在显卡驱动选项时选择no即可,其他项默认安装,安装路径选择默认路径

    安装好cuda之后,按住control+alt+F7进入图形界面,打开终端要添加环境变量:sudo gedit ~/.bashrc

    在最后一行添加

    export PATH=/usr/local/cuda-9.0/bin:$PATH

    export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64:$LD_LIBRARY_PATH

    点击保存之后退出,输入source  ~/.bashrc刷新一下环境,输入nvcc -V测试cuda是否安装成功

    特别注意:安装完cuda一定先别重启!!!

    安装cuda时会有一定概率重启时鼠标键盘不可用,去查了一下原因,是因为在安装cuda时可能会把xserver-xorg-input-all这个软件包给删掉,所以在重启系统之前一定要先把xserver-xorg-input-all重新安装回来,具体命令

    sudo apt install xserver-xorg-input-all 

    需要什么依赖软件,对应安装好即可

    3、安装cudnn

    cudnn的安装我采用了最简单的方式deb文件安装,点击这里跳转到下载页,选择cuda9.0版本的cudnn

    选择第456个文件

    选择ubuntu16对应的三个deb文件进行下载

    下载cudnn是要注册nvidia账号的,注册完账号后加入小组即可下载文件

    下载完成以后直接用sudo dpkg -i指令安装deb文件即可

    验证方法:

    cp -r /usr/src/cudnn_samples_v7/ $HOME

    cd $HOME/cudnn_samples_v7/mnistCUDNN

    make clean && make

    ./mnistCUDNN

    如果输出结果为Test passed!即说明已经安装成功了

    这时候重启就完成了!

    相关文章

      网友评论

        本文标题:Ubuntu16系统显卡安装以及CUDA,CUDNN的安装(Ge

        本文链接:https://www.haomeiwen.com/subject/cmjgpctx.html