我的搭配是:
tensorflow_gpu-1.0.1-cp27-none-linux_x86_64.whl
cudnn-8.0-linux-x64-v5.0-ga.tgz
cuda_8.0.61_375.26_linux-run
剛開始使用tensorflow1.1.0rc1,發現不行的
cudnn使用v5.1和之後的版本是不行的
注意在.bashrc中寫
export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64:/usr/local/cuda-8.0/extras/CUPTI/lib64:/usr/local/cuda/lib64:$LD_LIBRARY_PATH
#/usr/local/cuda是cudnn解壓的檔案
export CUDA_HOME=/usr/local/cuda-8.0
export PATH=/usr/local/cuda-8.0/bin:$PATH