Ubuntu 16.04 使用pip命令安装 tensorflow 1.7
本站教程收集整理的这篇文章主要介绍了Ubuntu 16.04 使用pip命令安装 tensorflow 1.7,本站教程本站觉得挺不错的,现在分享给大家,也给大家做个参考。
参考:
https://tensorflow.google.cn/install/install_linux
目的
基于python 3.5,采用pip的方式在Ubuntu 16.04 上安装tensorflow 1.7 GPU版本
环境要求:
Ubuntu 16.04 或更高版本
CUDA 9.0
cuDNN v7.0 (注意:不要安装v7.1)
主要步骤
一、安装显卡驱动
二、安装CUDA 9.0
三、安装cuDNN v7
下载并解压cuDNN v7,在终端输入以下命令执行安装(要注意当前终端的路径):
sudo cp -i include/cudnn.h /usr/local/cuda/include sudo cp -i lib64/libcudnn* /usr/local/cuda/lib64 sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn* sudo ldconfig /usr/local/cuda/lib64
四、采用pip命令安装
(1)安装pip3 软件包管理器
sudo apt-get install python3-pip python3-dev # for Python 3.n
(2)使用pip命令安装
sudo pip3 install tensorflow-gpu # Python 3.n; GPU support
(可选)如果提示“pip3 install tensorflow-gpu # Python 3.n; GPU support”,则先升级pip3 :
sudo pip3 install --upgrade pip
(3)验证安装
在 Python 交互式 sHell 中输入以下几行简短的程序代码:
# Python import tensorflow as tf Hello = tf.constant('Hello,TensorFlow!') sess = tf.SessiON() print(sess.run(Hello))
如果系统输出以下内容,就说明您可以开始编写 TensorFlow 程序了:
Hello,TensorFlow!
或
b'Hello,TensorFlow!'
五、查看tensorflow版本【可选】
import tensorflow as tf tf.__version__
故障处理:
(1) ImportError: libcublas.so.9.0: cAnnot open shared object file: No such file or directory
CUDA9.0未安装,或安装后路径未加到环境变量。
(2) Loaded runtime CuDNN library: 7103 (compatibility version 7100) but source was compiled with 7005 (compatibility version 7000)
cuDNN版本错误。当前cuDNN的版本是v7.1,应该安装v7.0。
本站总结
以上是本站教程为你收集整理的Ubuntu 16.04 使用pip命令安装 tensorflow 1.7全部内容,希望文章能够帮你解决Ubuntu 16.04 使用pip命令安装 tensorflow 1.7所遇到的程序开发问题。
如果觉得本站教程网站内容还不错,欢迎将本站教程推荐给好友。
本图文内容来源于网友网络收集整理提供,作为学习参考使用,版权属于原作者。
如您有任何意见或建议可联系处理。
《Ubuntu 16.04 使用pip命令安装 tensorflow 1.7》来自互联网同行内容,若有侵权,请联系我们删除!
相关文章
发表评论
评论列表
- 这篇文章还没有收到评论,赶紧来抢沙发吧~