Bump CUDA from 11.2 to 11.8

Update Pytorch, Torchvision and Tensorflow to use CUDA 11.8 builds. Also bumped tensorflow from 2.9.1 to 2.14.0 because lower versions <2.13.1 only has CUDA 11.2 on conda-forge.
pangeo-data · Dec 12, 2023 · 7922898 · 7922898
1 parent cf1f932
commit 7922898
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/ml-notebook/environment.yml b/ml-notebook/environment.yml
@@ -9,4 +9,4 @@ dependencies:
  - jax
  - jupyterlab-nvdashboard
  - keras-cv
- - tensorflow>=2.9.1=*cuda112*
+ - tensorflow>=2.14.0=*cuda118*
diff --git a/pytorch-notebook/environment.yml b/pytorch-notebook/environment.yml
@@ -7,6 +7,6 @@ channels:
 dependencies:
  - jupyterlab-nvdashboard
  - gpytorch
- - pytorch>=2.0.0=*cuda112*
- - torchvision>=0.15.1=*cuda112*
+ - pytorch>=2.0.0=*cuda118*
+ - torchvision>=0.15.1=*cuda118*
  - torchgeo