我正在尝试使用TensorFlow 1.x运行GitHub代码。我正在使用colab来做这个。我遇到了这样的问题,我找不到解决的办法。我使用的是TensorFlow1.15,我安装的Cuda版本是10.1,colab中的Nvidia驱动版本是NVIDIA-SMI450.51.05驱动版本: 418.67。当我尝试运行上面的另一段代码时,cuda似乎可以正常运行。我在colab上使用GPU模式,有人能帮帮我吗?谢谢。


发布于 2020-07-25 21:21:21
可能笔记本会话未连接到gpu,请尝试重新启动会话,然后等待笔记本分配所需的资源。
发布于 2020-07-28 18:01:15
请分享Colab的链接,因为我认为您没有将colab配置为使用GPU。您还可以按照以下步骤进行操作:
https://stackoverflow.com/questions/63088747
复制相似问题