Sagemaker Studio Lab GPU 不能與 pytorch 一起使用

Question

我無法讓 GPU 工作，即使我在開始時選擇了 GPU 選項 4 小時。 有什么辦法可以激活它嗎？

編輯：

我用 cuda 安裝了 pytorch： %conda install pytorch torchvision torchaudio cudatoolkit=11.4 -c pytorch

Answer 1

你可以嘗試通過 YAML 文件選項構建自定義環境，使用類似這樣的東西（下面只是一個適用於我的示例文件，你可以添加特定於你的用例的刪除包）

name: torch
channels:
    - pytorch
dependencies:
  - python=3.9
  - pip
  - pip:
    - ipywidgets
  - conda
  - conda:
    - ipykernel
    - pytorch
    - torchvision
    - torchaudio
    - torchserve
    - cudatoolkit=11.3
    - sklearn