远程计算机上带有 vscode 的 jupyter 笔记本在张量流中看不到 GPU

Question

我使用 SSH 在远程计算机上运行 vscode。 tensorflow环境有tensorflow 2.17、python 3.11，在vscode中，python windows检测GPU。 但是，在 vscode 的 Jupyter 窗口中运行时，检测 GPU 的相同 Python 代码将不会检测 GPU。

这个问题似乎与如何在jupyter笔记本中使用远程机器的GPU相同 但那里的解决方案不适用于我的情况。

我发现了这个：https://cschranz.medium.com/connect-vs-code-to-your-remote-gpu-jupyter-instance-58b86a195d9e

但我不明白。看起来我需要运行一个单独的 jupyter 实例，但同样，我不理解它，也不知道这是否能解决我的问题。

为什么 jupyter 窗口与 vscode 中的 python 窗口产生不同的结果？ 需要做什么才能让 vscode 中的 Jupyter 看到 GPU？

我当前的解决方法是在 python 窗口中编写代码，尽管这不如使用 Jupyter 单元方便。

Answer 1

您需要验证Jupyter笔记本内核与Python环境是否相同，然后您需要在Jupyter中激活正确的环境，如果未安装请使用以下命令

conda install jupyter or pip install jupyter

现在启动jupyter环境

jupyter notebook --no-browser --port=8888

还有一件事是，您必须确保 TensorFlow 环境（适用于 Python 脚本和 Jupyter Notebook）安装了正确版本的 CUDA 和 cuDNN。

这可能会帮助您查看 Jupyter 单元中的 GPU，就像运行 Python 脚本时一样。