1.安装tensorflow指定gpu版本:pip3 install --upgradehttps://storage.googleapis.com/tensorflow/linux/gpu/tensorflow_gpu-1.2.1-cp35-cp35m-linux_x86_64.whl
cpu版本为:pip installhttps://storage.googleapis.com/tensorflow/linux/cpu/tensorflow-1.2.1-cp35-cp35m-linux_x86_64.whl
如果是安装到管理员目录下,则前面加上sudo
2.tensorflow的cpu和gpu版本需要分别安装:
cpu版本:pip install tensorflow
gpu版本:pip installtensorflow-gpu
3.linux下运行gpu版tensorflow程序报
libcudnn.so.6.0: cannot open shared object file: No such file or directory错误,原因可能是路径问题,也可能是版本问题,比如我的是因为tensorflow1.3版本,而cudnn版本为5.0,所以造成文件可能找不到