在 AWS SageMaker pytorch-1.8-gpu-py36 實例上未檢測到 GPU

Question

我有一個在 AWS SageMaker Studio 上運行的 pytorch-1.8-gpu-py36 實例。

如果我在筆記本中輸入：

!nvidia-smi -L

我得到：

GPU 0: Tesla T4 (UUID: GPU-786d298a-2648-3506-6c3a-f541fa46d777)

但是如果我打開一個終端並輸入：

nvidia-smi -L

我找不到命令，如果我嘗試運行需要 GPU 的 .py 腳本，我會從 PyTorch 收到此錯誤：

pytorch_lightning.utilities.exceptions.MisconfigurationException: 
You requested GPUs: [0]
But your machine only has: []

即使終端 windows 和筆記本位於同一文件夾中，它們是否也會從不同的實例中運行？ 有沒有辦法讓終端與筆記本成為同一實例的一部分？

我不能簡單地從筆記本運行命令行，因為我需要一個無法從筆記本界面激活的 Conda 環境。

Answer 1

SageMaker Studio 的系統終端運行在與您的筆記本不同的實例上（請參閱 此處的 Studio 架構）。

要為運行筆記本的 GPU 實例使用終端，請單擊筆記本中的“圖像終端”圖標。 在選擇 PyTorch GPU 作為您的 SageMaker 圖像優化后，您還可以使用 New Launcher -> Image Terminal。