tensorflow使用的一些问题

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__



猜你喜欢

转载自blog.csdn.net/Lisatiy/article/details/80663566