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

可能是筆記本 session 沒有連接 gpu，嘗試重啟 session 並等待筆記本分配所需的資源。

Answer 2

請分享 Colab 的鏈接，因為我認為您沒有將 colab 配置為使用 GPU。 您也可以按照以下步驟操作，

1. Go 轉 Colab
2. 在左上角的編輯選項中
3. 在硬件加速器下拉菜單中，select GPU。