cudnn需要6.0以上
cuda的环境变量必须配置
sudo vim ~/.bashrc
export LD_LIBRARY_PATH=”$LD_LIBRARY_PATH:/usr/local/cuda/lib64”
export CUDA_HOME=/usr/local/cuda
source ~/.bashrc
安装指令为:
sudo pip install tensorflow-gpu==1.1.0 # Python 2.7; GPU support
sudo pip3 install tensorflow-gpu==1.1.0 # Python 3.n; GPU support
注意tensorflow版本不要超过1.4,否则需要cuda9.0
查询tensorflow的安装版本和路径:
python
import tensorflow as tf
tf.__version__
tf.__path__