加入收藏 | 设为首页 | 会员中心 | 我要投稿 李大同 (https://www.lidatong.com.cn/)- 科技、建站、经验、云计算、5G、大数据,站长网!
当前位置: 首页 > 综合聚焦 > 服务器 > Linux > 正文

LINUX实操:深度学习环境配置Ubuntu16.04+CUDA8.0+CUDNN5

发布时间:2020-12-13 17:53:20 所属栏目:Linux 来源:网络整理
导读:《LINUX实操:深度学习环境配置Ubuntu16.04+CUDA8.0+CUDNN5》要点: 本文介绍了LINUX实操:深度学习环境配置Ubuntu16.04+CUDA8.0+CUDNN5,希望对您有用。如果有疑问,可以联系我们。 深度学习从12年开始打响,配置深度学习环境软件一直是一个头疼的问题,如

《LINUX实操:深度学习环境配置Ubuntu16.04+CUDA8.0+CUDNN5》要点:
本文介绍了LINUX实操:深度学习环境配置Ubuntu16.04+CUDA8.0+CUDNN5,希望对您有用。如果有疑问,可以联系我们。

深度学习从12年开始打响,配置深度学习环境软件一直是一个头疼的问题,如何安装显卡驱动,如何安装CUDA,如何安装CUDNN;Ubuntu官方一直吐槽Nvidia显卡驱动有问题,网上大神也给出了关闭lightm巴拉巴拉之类的解决办法,有时候倒腾一星期也不见得安装成功,之前听一位师兄提起一种办法,然而在网上搜索也搜索不到,这里给出安装的教程,供大家参考

Ubuntu安装好,显卡插上去之后,一般的,配置pytorch,tensorflow,mxnet,caffe等深度学习环境必要做三件事情

1.安装NVIDIA驱动

通过敕令行安装,sudo apt-get install nvidia-375

2. 安装cuda toolkit

sudo sh cuda_8.0.61_375.26_linux.r7un.26_linux-run

因为在第一步已经安装NVIDIA驱动,所以在安装cuda的时候不消安装opengl,除此之外一路绿灯

3.安装cudnn

3.1 进入下载的cudnn/include,cd cudnn/include, 拷贝cudnn.h 到/user/local/cuda-8.0/include/

sudo cp cudnn.h /user/local/cuda-8.0/include/

3.2 进入lib64文件夹,拷贝lib_cudnn_static.a 到/user/local/cuda-8.0/lib64文件夹下,sudo cp lib_cudnn_static.a?/user/local/cuda-8.0/lib64/

3.3 拷贝libcudnn.so.5.1.3到/user/local/cuda-8.0/lib64/下,sudo cp?libcudnn.so.5.1.3 /user/local/cuda-8.0/lib64/

3.4 进入/user/local/cuda-8.0/lib64/下而且创建软链接,sudo ln -s libcudnn.so.5.1.3 libcudnn.so.5,sudo ln -s libcudnn.so.5 libcudnn.so

至此完毕

下面是之前的记载

本文永远更新链接地址

linux

学习更多LINUX教程,请查看站内专栏,如果有LINUX疑问,可以加QQ交流咨询。

(编辑:李大同)

【声明】本站内容均来自网络,其相关言论仅代表作者个人观点,不代表本站立场。若无意侵犯到您的权利,请及时与联系站长删除相关内容!

    推荐文章
      热点阅读