Pytorch not importing

[dweisz]

Bug description

pytorch is installed on astro.datahub. However, we try to import it, we get the below error (see screenshot). Re-installing pytorch via conda or pip doesn't seem to resolve the issue. Any guidance is appreciated.

Environment & setup

• Hub: AstroHub
• Language: Python

How to reproduce

import torch

[shaneknapp]

looking in to this now...

…

On Thu, Apr 4, 2024 at 12:00 PM Dan Weisz ***@***.***> wrote: Assigned #5659 <#5659> to @shaneknapp <https://github.com/shaneknapp>. — Reply to this email directly, view it on GitHub <#5659 (comment)>, or unsubscribe <https://github.com/notifications/unsubscribe-auth/AAMIHLHKOA7F75IBGIF5LI3Y3WPLJAVCNFSM6AAAAABFX2RNCWVHI2DSMVQWIX3LMV45UABCJFZXG5LFIV3GK3TUJZXXI2LGNFRWC5DJN5XDWMJSGM3DGMZQHEZDONQ> . You are receiving this because you were assigned.Message ID: ***@***.*** com>

[shaneknapp]

hmm, looks like this is a pytorch issue: pytorch/pytorch#123097

i'll see if i can't make this work somehow.

[shaneknapp]

@dweisz could you log in to astro-staging.datahub.berkeley.edu and confirm that it works for you?

[dweisz]

Unfortunately, it's still throwing me an error

[balajialg]

@dweisz Are you logged into https://astro-staging.datahub.berkeley.edu? It works without any errors in Astro-staging hub for me. I can't see your username drweisz listed in the admin interface of staging hub (It should have if you had logged into that hub)

[dweisz]

Apologies, I was logged into the wrong server. Yes, import torch appears to work on the astro-staging instance.

…

On Apr 4, 2024, 1:47 PM -0700, Balaji Alwar ***@***.***>, wrote: @dweisz Are you logged into https://astro-staging.datahub.berkeley.edu? It works without any errors in Astro-staging hub for me. image.png (view on web) — Reply to this email directly, view it on GitHub, or unsubscribe. You are receiving this because you were mentioned.Message ID: ***@***.***>

[shaneknapp]

Apologies, I was logged into the wrong server. Yes, import torch appears to work on the astro-staging instance.
…
On Apr 4, 2024, 1:47 PM -0700, Balaji Alwar @.>, wrote: @dweisz Are you logged into https://astro-staging.datahub.berkeley.edu? It works without any errors in Astro-staging hub for me. image.png (view on web) — Reply to this email directly, view it on GitHub, or unsubscribe. You are receiving this because you were mentioned.Message ID: @.>

glad this works! we'll get this merged to prod by EOD today

[ElHouas]

Hello @shaneknapp ,

I am experiencing the same issue when installing torch using conda inside a docker container using as base images ubuntu:22 and anaconda official docker image. Here is my env.yaml file:

name: ml_env
channels:

• pytorch
• nvidia
• conda-forge
• nodefaults
  dependencies:
• python=3.10.7
• mamba
• pip
• poetry=1.8.2
• pytorch=2.0.1
• torchaudio=2.0.2
• torchvision=0.15.2
• pytorch-cuda=11.8
• cudatoolkit=11.8.0
• cudnn
• blas=1.0
  platforms:
• linux-64

Thanks for the help! :)