unbuntu配置深度学习环境(cuda9.1+cudnn7.1+tensorflow-1.10.1)图文并茂 超详细

版权声明:本文为博主原创文章,遵循 CC 4.0 by-sa 版权协议,转载请附上原文出处链接和本声明。
本文链接:https://blog.csdn.net/weixin_38243916/article/details/89763737

听了dwsun的安装教学视频,在我自己安装的时候做一下记录,好记性不如烂笔头。

ubuntu 18.04 LTS    显卡是1060

在安装之前要先进行换源,aliyun的还是挺快的。

sudo apt-get update

sudo apt-get upgrade

sudo reboot

可以对系统进行易用性的设置,安装  git rsync zsh aptitude  等

sudo apt-get install git rsync zsh aptitude 

安装nvidia驱动

sudo aptitude install nvidia-384 --without-recommends

不同版本的号应该是不同的,安的时候去官网查一下。使用--without-recommends可以只安装需要的东西

安装完成后重启,在终端输入  nvidia-smi可查看是否安装成功,下面是我安装成功后的截图

下载cudnn

18.04的cuda不用下载,在Ubuntu官方的repo里自带cudn9.1。会安装在/usr/lib下面

下载cudnn,需要一个nvidia的开发账号,可以去官网下载。链接

我下载的是 cudnn-7.1-linux-x64-v7.1

安装cuda

直接在终端执行安装即可

sudo aptitude install nvidia-cuda-toolkit --without-recommends

输入y确认安装即可,这个我安的时候需要等一会,毕竟要下载1546MB,漫长的等待及优美的风扇声。

安装之后的样子

有兴趣可以看一下都安装在了哪里

cd /

sudo find . -type f -name *cuda*

sudo find . -type f -name *cudnn*

安装cudnn

tar -zxf  cudnn-9.1-linux-x64-v7.1.tgz

cd cuda

mkdir lib/x86_64-linux-gnu -p

mv lib64/* lib/x86_64-linux-gnu

rm -rvf lib64

sudo rsync -avp ./ /usr

安装tensorflow

可以在tinymind上下载非官方的支持cuda9.1和cudnn7.1的编译版,找到自己需要的额版本即可。

我下载的是这个版本   tensorflow-1.10.1-cp36-cp36m-linux_x86_64.whl

如果电脑的CPU型号比较老,可以考虑安装上面链接中的 generic版本

下载之后重新打开终端

安装pip

sudo aptitude install python3-pip

安装tensorflow

pip3 install tensorflow-1.10.1-cp36-cp36m-linux_x86_64.whl

在终端tensorflow包的路径下安装

安装完毕之后显示成功安装后,进入python3进行测试,输入 import  tensorflow as tf 不报错即为成功,有的如果默认的Python环境是Python2的话记得切换成Python3

展开阅读全文

没有更多推荐了,返回首页