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