linux安裝tensorflow搭配那個版本的cuda和cudnn?

l2xBrain發表於2017-04-13

我的搭配是:

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

相關文章