Merge pull request #5132 from berkeley-dsep-infra/staging

merging 5131 to prod
berkeley-dsep-infra · Oct 12, 2023 · d391cf7 · d391cf7
2 parents e05cfa6 + d0d2d11
commit d391cf7
Showing 1 changed file with 6 additions and 0 deletions.
diff --git 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
@@ -33,6 +34,11 @@ 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
+- torchvision==0.16.0
+- torchaudio==2.1.0
+- cpuonly==2.0
 - pip
 - pip:
   # - -r infra-requirements.txt