深度学习环境配置Ubuntu16.04+CUDA8.0+CUDNN5

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

成都创新互联网络公司拥有10余年的成都网站开发建设经验,成百上千家客户的共同信赖。提供网站建设、网站制作、网站开发、网站定制、买友情链接、建网站、网站搭建、成都响应式网站建设公司、网页设计师打造企业风格,提供周到的售前咨询和贴心的售后服务

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

至此完毕

下面是之前的记录

新闻名称:深度学习环境配置Ubuntu16.04+CUDA8.0+CUDNN5
路径分享:http://www.mswzjz.com/qtweb/news30/185730.html

网站建设、网络推广公司-创新互联,是专注品牌与效果的网站制作,网络营销seo公司;服务项目有等

广告

声明:本网站发布的内容(图片、视频和文字)以用户投稿、用户转载内容为主,如果涉及侵权请尽快告知,我们将会在第一时间删除。文章观点不代表本网站立场,如需处理请联系客服。电话:028-86922220;邮箱:631063699@qq.com。内容未经允许不得转载,或转载时需注明来源: 创新互联