Sagemaker Studio Lab GPU 无法与 pytorch 一起使用

【问题标题】：Sagemaker Studio Lab GPU not working with pytorchSagemaker Studio Lab GPU 无法与 pytorch 一起使用
【发布时间】：2022-02-25 06:53:57
【问题描述】：

即使我在开始时选择了 GPU 选项 4 小时，我也无法让 GPU 工作。有什么办法可以激活它吗？

编辑：

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

【问题讨论】：

标签： pytorch amazon-sagemaker

【解决方案1】：

您能否尝试通过 YAML 文件选项构建自定义环境，使用类似的东西（下面只是一个适用于我的示例文件，您可以添加特定于您的用例的删除包）

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

【讨论】：