TensorFlow笔记(一)--ubuntu系统安装TensorFlow

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/zby1001/article/details/53260076
ubuntu/Linux下安装:
仅使用CPU:
$ pip install https://storage.googleapis.com/tensorflow/linux/cpu/tensorflow-0.5.0-cp27-none-linux_x86_64.whl
仅使用GPU:
$ pip install https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow-0.5.0-cp27-none-linux_x86_64.whl
启用GPU支持:
需要设置LD_LIBRARY_PATHCUDA_HOME环境变量。
假定CUDA安装目录为usr/local/cuda,将下面两个命令添加到~/.bash_profile文件中:
export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/usr/local/cuda/lib64"
export CUDA_HOME=/usr/local/cuda

训练自己的第一个TensorFlow神经网络:
从源代码的根路径执行:

$cd tensorflow/models/image/mnist
$python convolutional.py
Succesfully downloaded train-images-idx3-ubyte.gz 9912422 bytes.
Succesfully downloaded train-labels-idx1-ubyte.gz 28881 bytes.
Succesfully downloaded t10k-images-idx3-ubyte.gz 1648877 bytes.
Succesfully downloaded t10k-labels-idx1-ubyte.gz 4542 bytes.
Extracting data/train-images-idx3-ubyte.gz
Extracting data/train-labels-idx1-ubyte.gz
Extracting data/t10k-images-idx3-ubyte.gz
Extracting data/t10k-labels-idx1-ubyte.gz
...........
...........
能正常运行说明安装成功

猜你喜欢

转载自blog.csdn.net/zby1001/article/details/53260076