TensorFlow 和 Cuda 不兼容

Question

我正在尝试使用 TensorFlow 1.x 运行 GitHub 代码。 我正在为此使用 colab。我遇到了这种我找不到解决方案的问题。 我使用的是 Tensorflow 1.15，我安装的 Cuda 版本是 10.1，colab 中的 Nvidia 驱动器版本是 NVIDIA-SMI 450.51.05 驱动程序版本：418.67。 当我尝试运行上面的另一个代码时，cuda 似乎可以正常工作。 我在 colab 上使用 GPU 模式 有人能帮助我吗？ 谢谢。

Answer 1

可能 notebook session 没有连接 gpu，尝试重启 session 并等待 notebook 分配所需资源。

Answer 2

请分享 Colab 的链接，因为我认为您没有将 colab 配置为使用 GPU。您也可以按照以下步骤操作，

1. 转到 Colab
2. 在左上角的编辑选项中
3. 在硬件加速器下拉菜单中，选择 GPU。