tensorflow - 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 模式 有人可以帮帮我吗？谢谢。

标签： tensorflowdeep-learningnvidia

Answer 1

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

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