1. 急速安装
如果没有科学上网工具,使用官方方式安装Tensorflow极其缓慢,解决方法是:使用国内源,这里使用豆瓣源。
pip install tensorflow-gpu -i https://pypi.douban.com/simple
但是安装过程中我这里报了一个错:“未成功卸载wrapt”。
ERROR: Cannot uninstall 'wrapt'. It is a distutils installed project and thus we cannot accurately determine which files belong to it which would lead to only a partial uninstall.
解决方式是跳过对wrapt的卸载安装:
pip install tensorflow-gpu -i https://pypi.douban.com/simple --ignore-installed wrapt
2. 测试是否安装成功,查看Tensorflow的版本和安装位置:
yhx@yhx:~$ python
Python 3.7.0 (default, Jun 28 2018, 13:15:42)
[GCC 7.2.0] :: Anaconda, Inc. on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import tensorflow as tf
>>> tf.__version__
'2.1.0'
>>> tf.__path__
['/home/yhx/conda/lib/python3.7/site-packages/tensorflow']
>>>