linux安装tensorflow搭配那个版本的cuda和cudnn？

2024年1月11日 146次阅读 来源: l2xBrain

我的搭配是:

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
    原文地址: https://segmentfault.com/a/1190000009056627
    本文转自网络文章，转载此文章仅为分享知识，如有侵权，请联系博主进行删除。
点赞