Ubuntu17.04安装TensorFlow1.2的GPU版本
来源:互联网 发布:sql查询分析器区别 编辑:程序博客网 时间:2024/06/06 02:45
摘要:Ubuntu17.04安装TensorFlow1.2的GPU版本。首先验证nvidia显卡,然后安装CUDA Toolkit 8.0,安装cuDNN v5深度神经网络计算加速库,最后通过python pip安装TensorFlow-GPU版本。
1.安装NVIDIA的GPU-CUDA,cuDNN
1.1. 查看当前配置
xiaolei@wang:~$ uname -m && cat /etc/*release
1.2.安装CUDA
【CUDA官方下载】https://developer.nvidia.com/cuda-downloads
xiaolei@wang:~/Downloads$ sudo dpkg -i cuda-repo-ubuntu1604-8-0-local-ga2_8.0.61-1_amd64.deb
xiaolei@wang:~/Downloads$ sudo dpkg -i cuda-repo-ubuntu1604-8-0-local-cublas-performance-update_8.0.61-1_amd64.deb
xiaolei@wang:~/Downloads$ sudo apt update
xiaolei@wang:~/Downloads$ sudo apt install cuda
1.3.安装cuDNN
NVIDIA cuDNN是用于深度神经网络的GPU加速库。
【原文】The NVIDIA CUDA® Deep Neural Network library (cuDNN) is a GPU-accelerated library of primitives for deep neural networks. cuDNN provides highly tuned implementations for standard routines such as forward and backward convolution, pooling, normalization, and activation layers. cuDNN is part of the NVIDIA Deep Learning SDK.
【官方下载】(需要注册登录后填写个小问卷才能下载)https://developer.nvidia.com/cudnn
- tensorflow-gpu r1.2现在(201707)默认的是cuDNN v5,而v6会报错。
- 安装方式很简单,下载后解压,把cudnn中的文件内容拷贝到对应的cuda中。
xiaolei@wang:~/Downloads$ tar -zxf cudnn-8.0-linux-x64-v5.1.tgz && cd cuda
xiaolei@wang:~/Downloads/cuda$ sudo cp include/cudnn.h /usr/local/cuda-8.0/include/
xiaolei@wang:~/Downloads/cuda/lib64$ sudo cp lib64/libcudnn* /usr/local/cuda-8.0/lib64
1.4.安装 libcupti-dev 包
xiaolei@wang:~$ sudo apt-get install libcupti-dev
1.5.变量设置
#!/bin/sh# Author:wangxiaolei 王小雷# Blog: http://blog.csdn.net/dream_an# Github: https://github.com/wangxiaoleiai# Date: 201707# Organization: https://github.com/whaleaiexport CUDA_HOME=/usr/local/cuda-8.0export PATH=$CUDA_HOME/bin:$PATHexport LD_LIBRARY_PATH=$CUDA_HOME/lib64
2.安装tensorflow-gpu
可以通过python2或者python3的安装,博主使用的是python3。
2.1.安装依赖
xiaolei@wang:~$ sudo apt-get install python-pip python-dev # for Python 2.7xiaolei@wang:~$ sudo apt-get install python3-pip python3-dev # for Python 3.n
2.2.安装tensorflow-gpu版本
xiaolei@wang:~$ pip install tensorflow-gpu # Python 2.7; GPU supportxiaolei@wang:~$ spip3 install tensorflow-gpu # Python 3.n; GPU support
3.测试tensorflow-gpu
xiaolei@wang:~$ python3
import tensorflow as tfhello = tf.constant('Hello, TensorFlow!')sess = tf.Session()print(sess.run(hello))
可以看到已经使用GPU了!完结-成功
1.彩蛋-卸载方法
xiaolei@wang:~$ sudo pip uninstall tensorflow # for Python 2.7xiaolei@wang:~$ sudo pip3 uninstall tensorflow # for Python 3.n
2.彩蛋-官网给出的常见错误处理方法
https://www.tensorflow.org/install/install_sources#common_installation_problems
- Ubuntu17.04安装TensorFlow1.2的GPU版本
- ubuntu17.04安装tensorflow1.3(GPU)详解
- Ubuntu17.04 CUDA8.0 Cudnn v7 tensorflow1.3-GPU安装
- Ubuntu16.04 配置tensorflow1.0 gpu版本
- Ubuntu16.04 配置tensorflow1.0 gpu版本
- Ubuntu16.04安装gpu版tensorflow1.2+cuda+cudnn
- Tensorflow1.0+ubuntu14.04+GPU安装
- Win10 配置tensorflow1.3 GPU版本以及tensorboard的使用
- Tensorflow1.4安装gpu教程
- 双系统安装 windows10 ubuntu17.04server版本
- ubuntu16.04安装gpu版本的tensorflow
- Ubuntu16.04+cuda-8.0+cudnn-v5.1+tensorflow0.8-gpu/tensorflow1.0-gpu安装教程
- Windows 10 下安装Tensorflow1.0.1-GPU
- Win10 TensorFlow1.4 (gpu)安装
- win10下安装TensorFlow1.4.0-GPU
- Ubuntu17.04+cuda8.0+cudnn6.0+tensorflow1.3配置
- Ubuntu17.04+1080ti+cuda9+cudnn7+tensorflow1.4/1.3配置
- VMware Ubuntu17.04(64位)下安装CPU版本的Caffe
- 计算机网络自顶向下 :应用层(FTP、SMTP、POP3、IMAP)
- JAVA字符串格式化-String.format()的使用
- XTabLayout
- java入门网站
- web.xml配置详解之监听器,过滤器
- Ubuntu17.04安装TensorFlow1.2的GPU版本
- Hibernate的事务处理机制和flush方法的用法
- JAVA8-Consumer函数
- 十进制和任意进制的转换
- 个案管理(Case management)
- Android MTK选择重启,可以完整播放完关机动画
- Leetcode——639.Decode Ways II
- Screen 操作快捷键
- HDU 3874__Necklace