From bcdd5576d896750f32d60a512482d0edf3fb0d41 Mon Sep 17 00:00:00 2001 From: shane knapp Date: Thu, 12 Oct 2023 13:54:21 -0700 Subject: [PATCH 1/3] adding pytorch --- deployments/data102/image/environment.yml | 2 ++ 1 file changed, 2 insertions(+) diff --git a/deployments/data102/image/environment.yml b/deployments/data102/image/environment.yml index 820ccc18a..db8b27f90 100644 --- a/deployments/data102/image/environment.yml +++ b/deployments/data102/image/environment.yml @@ -33,6 +33,8 @@ dependencies: - scipy==1.10.1 - seaborn==0.12.2 - statsmodels==0.14.0 +# fall 2023: https://github.com/berkeley-dsep-infra/datahub/issues/5130 +- pytorch==2.1.0 - pip - pip: # - -r infra-requirements.txt From 8a57808004a507715255c776071aac50808d4f3c Mon Sep 17 00:00:00 2001 From: shane knapp Date: Thu, 12 Oct 2023 14:09:27 -0700 Subject: [PATCH 2/3] adding pytorch channel --- deployments/data102/image/environment.yml | 1 + 1 file changed, 1 insertion(+) diff --git a/deployments/data102/image/environment.yml b/deployments/data102/image/environment.yml index db8b27f90..e3b26dd13 100644 --- a/deployments/data102/image/environment.yml +++ b/deployments/data102/image/environment.yml @@ -2,6 +2,7 @@ name: data102-FA23 channels: - conda-forge +- pytorch dependencies: - python==3.11.3 From 555d742c945f86304ba11f15e77982fafbbd558d Mon Sep 17 00:00:00 2001 From: shane knapp Date: Thu, 12 Oct 2023 14:15:59 -0700 Subject: [PATCH 3/3] adding other recommended pkgs --- deployments/data102/image/environment.yml | 3 +++ 1 file changed, 3 insertions(+) diff --git a/deployments/data102/image/environment.yml b/deployments/data102/image/environment.yml index e3b26dd13..040433b3f 100644 --- a/deployments/data102/image/environment.yml +++ b/deployments/data102/image/environment.yml @@ -36,6 +36,9 @@ dependencies: - statsmodels==0.14.0 # fall 2023: https://github.com/berkeley-dsep-infra/datahub/issues/5130 - pytorch==2.1.0 +- torchvision==0.16.0 +- torchaudio==2.1.0 +- cpuonly==2.0 - pip - pip: # - -r infra-requirements.txt