ubuntu安装TensorFlow-GPU教程

本文的安装环境是:ubuntu16.04、CUDA8.0、Cudnn6.0,tensorflow1.3
在这里提供下面所需要的安装文件供大家使用
链接:http://pan.baidu.com/s/1c2atLMo 密码:78qm

下面的命定不要直接复制粘贴,要根据个人文件路径的实际情况。

安装NVIDIA驱动

先去官网查看适合自己GPU版本的驱动:http://www.nvidia.com/Download/index.aspx?lang=en-us
以GTX1080Ti为例:查到其对应的版本号为:384



得到版本号之后就是执行下面的命定来更新显卡驱动:

sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt-get update
sudo apt-get install nvidia-(自己的版本号)
sudo apt-get install mesa-common-dev
sudo apt-get install freeglut3-dev

执行成功后,重启电脑,打开终端,输入
nvidia-smi
如果没有报错并且能够显示,则说明你的驱动更新成功了。

安装CUDA

先去下载CUDA8.0的安装文件https://developer.nvidia.com/cuda-80-ga2-download-archive
然后再执行以下命定

sudo dpkg -i cuda-repo-ubuntu1604-8-0-local-ga2_8.0.61-1_amd64.deb
sudo dpkg -i cuda-repo-ubuntu1604-8-0-local-cublas-performance-update_8.0.61-1_amd64.deb
sudo apt-get update
sudo apt-get install cuda
sudo dpkg -i cuda-repo-ubuntu1604-8-0-local-ga2_8.0.61-1_amd64.deb
sudo dpkg -i cuda-repo-ubuntu1604-8-0-local-cublas-performance-update_8.0.61-1_amd64.deb
sudo apt-get update 

到这里就装完了CUDA

安装Cudnn

下载地址:https://developer.nvidia.com/cudnn
再执行命定

tar -zxf cudnn-8.0-linux-x64-v6.0.tgz
cd cuda
sudo cp include/cudnn.h /usr/local/cuda-8.0/include/
sudo cp lib64/libcudnn* /usr/local/cuda-8.0/lib64
tar -zxf cudnn-8.0-linux-x64-v6.0.tgz
cd cuda
sudo cp include/cudnn.h /usr/local/cuda-8.0/include/
sudo cp lib64/libcudnn* /usr/local/cuda-8.0/lib64 

安装Anconada

下载Anconada文件
然后执行命定
bash (Anconada的安装文件名)

安装TensorFlow

找到对应版本,然后执行命定
(在这之前可能要测试下自己的电脑上安装了pip没,如果没有的话,请安装)

pip install (TensorFlow的安装文件)

然后就可以在自己机子上测试是否安装成功了
先在终端进入python
然后输入

import tensorflow as tf

如果没报错,则说明安装成功: congratulations!

参考文献

感谢前辈的帖子
http://www.linuxidc.com/Linux/2017-01/139321.htm
http://118.89.51.133/?p=41

番外

本人在安装的过程中发现,在上面的步骤进行到安装Anconada后
直接输入命定:

conda install tensorflow-gpu

也能成功安装
另外安装keras也能用类似命定

conda install keras-gpu

只能说Anconada真的很强大

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
【社区内容提示】社区部分内容疑似由AI辅助生成,浏览时请结合常识与多方信息审慎甄别。
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

相关阅读更多精彩内容

友情链接更多精彩内容