安装
pip install tensorflow
如果安装GPU版本的,需要安装 Cuda Toolkit 8.0 和 cuDNN v5.1
pip install tensorflow-gpu
并且设置环境变量LD_LIBRARY_PATH
和CUDA_HOME
。运行 demo model
python -m tensorflow.models.image.mnist.convolutional
pip install tensorflow
如果安装GPU版本的,需要安装 Cuda Toolkit 8.0 和 cuDNN v5.1
pip install tensorflow-gpu
并且设置环境变量 LD_LIBRARY_PATH
和CUDA_HOME
。
运行 demo model
python -m tensorflow.models.image.mnist.convolutional