Tensorflow GPU导入错误

时间:2018-02-24 23:23:29

标签: tensorflow

我有CUDA 8.0,我可以下载cuDNN。目前,我有适用于Linux的cuDNN版本7.0.5。

我没有管理员权限。

当我尝试为GPU安装TensorFlow版本1.4时,我收到了此错误:

 ImportError: libcudnn.so.6: cannot open shared object file: No such file or directory

我认为这是因为我的机器上没有cuDNN。我根据sysadmin的建议下载了7.0.5版本,这当然不是错误消息要求我获得的版本(它需要版本6)。

所以我想,我会尝试使用Tensorflow 1.5版的GPU。我收到了这个错误:

ImportError: libcublas.so.9.0: cannot open shared object file: No such file or directory

我该怎么办?有没有办法下载旧版本的cuDNN?或者在某处下载cublas 9.0的方法?

1 个答案:

答案 0 :(得分:0)

如果您在nvidia注册,也可以下载旧版本的cuDNN。虽然它有点隐藏。确保下载与您的cuda版本兼容的正确版本。另外,不要忘记为tensorflow设置CUDA_HOME环境变量以找到您的GPU。