ubuntu安装TensorFlow

2019-04-05  本文已影响0人  Toplhyi

使用Python语言作为开发语言。

  1. 安装Anaconda
bash Anaconda3-5.3.1-Linux-x86_64.sh
  1. 安装Bazel
sudo add-apt-repository ppa:webupd8team/java
sudo apt update
sudo apt install oracle-java8-installer
sudo apt install pkg-config zip g++ zlib1g-dev unzip
chmod +x bazel-0.24.0-installer-linux-x86_64.sh
./bazel-0.24.0-installer-linux-x86_64.sh --user
sudo apt install python3-numpy swig python3-dev python3-wheel

Bazel安装完成后,会在home空间下生成名为bin的文件夹,里面是一个名为bazel的脚本文件。将bzael导入PATH:

export PATH="$PATH:$HOME/bin"

在终端输入bazel命令验证是否安装成功。
ps:安装时出现了如下错误,原因是bazel安装包所在路径包含中文。

Error: LinkageError occurred while loading main class com.google.devtools.build.lib.bazel.Bazel
    java.lang.ExceptionInInitializerError: null
  1. 安装CUDA。
    CUDA是NVIDIA推出的使用GPU资源进行复杂通用并行计算的SDK。
    可以在CUDA GPUs上查看自己显卡的计算能力值。
    显卡
    显卡计算能力值
make all

如果make成功,会在最后提示“Finished building CUDA samples”信息。之后,在/usr/local/cuda/extras/demo_suite文件夹下,可以找到可执行文件deviceQuery并在终端运行该文件,会输出GPU相关信息。
折腾了几天,并且重装了两次系统之后还是没有搞定,最终决定暂时放弃GPU。所以第4步跳过。

  1. 安装cuDNN。
tar -zxvf cudnn-10.1-linux-x64-v7.5.0.56.tgz

解压后会在当前目录生成一个名为cuda的文件夹,进入该文件夹。

sudo cp lib64/libudnn* /usr/local/cuda/lib64/
sudo cp include/duDNN.h /usr/local/cuda/include/
cd /usr/local/cuda/lib64/
sudo chmod +r libcuDNN.
export   LD_LIBRARY_PATH=/usr/local/cuda-10.1/lib64:usr/local/cuda-10.1/extras/CPUTI/lin64:$LD_LIBRARY_PATH
export CUDA_HOME=/usr/local/cuda-10.1
export PATH=/usr/local/cuda-10.1/bin:$PATH

之后执行source ~/.bashrc命令使之生效。

  1. 正式安装TensorFlow(使用pip安装)
pip install --upgrade tensorflow
  1. 测试TensorFlow
    输入一下代码,输出结果:array([4,6], dtype=float32)
import tensorflow as tf
a = tf.constant([1.0,2.0],name='a')
b = tf.constant([3.0,4.0],name='b')
result = a + b
sess = tf.Session()
sess.run(result)

如图:


测试
上一篇下一篇

猜你喜欢

热点阅读