我的搭配是:

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

l2xBrain
292 声望6 粉丝

Done is better than perfect


引用和评论

0 条评论