studio-lab-examples icon indicating copy to clipboard operation
studio-lab-examples copied to clipboard

Published 20 hours ago • verified publisher icon aws

Example pytorch env

Open robmarkcole opened this issue 3 years ago • 5 comments

As title, with GPU dependencies installed. Thanks

robmarkcole avatar Feb 07 '22 11:02 robmarkcole

The below recipe appears fine:

name: pytorch
dependencies:
  - python=3.9
  - pip
  - pip:
    - ipywidgets
  - conda
  - conda:
    - ipykernel
    - pytorch 
    - torchvision

robmarkcole avatar Feb 07 '22 12:02 robmarkcole

Thanks Robin for sharing the yaml example, could you provide more details about the issue you have?

ioanfr avatar Feb 16 '22 13:02 ioanfr

Hi @ioanfr this is not an issue as such, rather a feature request, for a validated env - note I did not test GPU access etc

robmarkcole avatar Feb 16 '22 14:02 robmarkcole

The validation environment will depend on libraries and framework used by each user. You can find more sample environments here. Also we will continue to add more sample configuration files to create various environments.

ioanfr avatar Feb 18 '22 15:02 ioanfr

An issue I encountered with this env: AttributeError: module 'torch._C' has no attribute '_cuda_setDevice' covered in https://github.com/pytorch/pytorch/issues/33655

robmarkcole avatar Mar 15 '22 14:03 robmarkcole