Блог

Я пытаюсь использовать последнюю версию PyTorch (1.7.0) в среде Conda на Sagemaker, указав версию пакета в файле environment.yml. Однако я получаю ошибку ResolvePackageNotFound. Обратите внимание, что я просто работаю в ноутбуке Jupyter с ядром, соответствующим этой среде Conda. Я не использую образ глубокого обучения.

Шаги по воспроизведению:

Сохраните приведенный ниже код в виде файла .yml, перейдите к тому, где это .файл yml сохраняется и запускается conda env create environment.yml .

 name: test_env
dependencies:
  - numpy
  - pandas
  - pytorch>=1.7.0
  - torchvision
  - scipy
  - ipykernel
  - torchvision
 

Я пробовал это на экземплярах разных типов (ml.p2.xlarge, ml.p3.2xlarge и ml.p3.8xlarge) и каждый раз получал одну и ту же ошибку. Я также пробовал это с версиями conda 4.8.3 и 4.9.2. Если я укажу pytorch>=1.5.0 , я смогу успешно создать среду.

У кого-нибудь есть идеи, почему я не могу успешно создать среду с более свежими версиями PyTorch? Основываясь на этой документации, я задаюсь вопросом, предустановил ли Sagemaker определенные версии PyTorch, и что-то идет не так, когда я пытаюсь использовать более свежую версию.

Экземпляры Sagemaker не всегда поддерживают последние пакеты. Проверьте эту ссылку для получения списка поддерживаемых образов в экземплярах Sagemaker.