0

我正在尝试在启用了 GPU 的谷歌云 vm 实例中运行 tensorflow python 脚本。我遵循了安装 GPU 驱动程序、cuda、cudnn 和 tensorflow 的过程。但是,每当我尝试运行我的程序(在超级计算集群中运行良好)时,我都会不断得到:

未定义符号:cudnnCreate

我已经在我的 ~/.bashrc 旁边添加了

export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/usr/local/cuda-8.0/lib64:/usr/local/cuda-8.0/extras/CUPTI/lib64:/usr/local/cuda-8.0/lib64"

导出 CUDA_HOME="/usr/local/cuda-8.0" 导出 PATH="$PATH:/usr/local/cuda-8.0/bin"

但它仍然不起作用并产生相同的错误

4

1 回答 1

1

回答我自己的问题:问题不是没有安装库,安装的库是错误的版本,因此找不到它。在这种情况下,它是 cudnn 5.0。但是,即使安装了正确的版本,由于驱动程序、CUDA 和 cudnn 版本之间的不兼容,它仍然无法工作。我通过重新安装包括驱动程序在内的所有内容来解决所有这些问题,同时考虑到 tensorflow 库的要求。

于 2017-04-05T19:14:56.207 回答