SSH远程通过Vscode运行Jupyter无法使用CUDA

1.问题描述

VSCode 远程 Jupyter Notebook 中 torch.cuda.is_available() 为 False，但终端同一环境运行脚本显示 True，导致 notebook 里报 "GPU is not available"。

运行如下代码时报错：

raise ValueError("GPU is not available. Change Colab runtime.")

原因如下：

• Notebook 使用的是正确的 Python 解释器，但 Jupyter 服务器进程未在可见 GPU 的会话/节点上启动，或 VSCode 连接的 Jupyter Server 不是 GPU 会话。
• 环境里缺少 jupyterlab，无法直接启动 Jupyter 服务。

2.解决方法

2.1安装Jupyter服务器包

在VScode重新新建一个终端，在终端安装安装jupyter lab：

conda install -n pytorch_gpu -c conda-forge jupyterlab -y

2.2 在 GPU 可见的终端里启动：

（1）激活 pytorch环境

conda activate pytorch_gpu

（2）在 GPU 可见的 shell 里重新启动 Jupyter

CUDA_VISIBLE_DEVICES=1 jupyter lab --no-browser --port=8888

2.3重新启动Jupyter的kernel

（1）首先复制上图中链接：

http://127.0.0.1:8888/lab?token=fd6d2be80d4a1bd497301605e5780dd93ee1d420632a5525

（2）然后选择新的内核

（3）选择Jupyter服务器

（4）连接到服务器

将上面远程URL的链接复制到下面：

2.4运行成功