安装Tensorflow GPU时出错

Question

我之前使用的是仅CPU版本的tensorflow。我现在尝试使用this链接安装GPU版本。

但我觉得我搞砸了。

当我尝试import tensorflow时，它会显示以下消息：

ImportError: libcudart.so.7.5: cannot open shared object file: No such file or directory

我该怎么办？

Answer 1

可能是因为安装了错误版本的cuda：检查export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/usr/local/cuda/lib64" export CUDA_HOME=/usr/local/cuda 以了解已安装的cuda版本以及它是否与Tensorflow错误中的版本匹配。如果两个版本都不匹配，则您必须安装另一个版本的cuda或Tensorflow。

另一个原因可能是因为缺少环境变量（如here所述）。 试试这个：

<%= render 'my_partial', catgories: @categories %>