diff --git a/ml-notebook/environment.yml b/ml-notebook/environment.yml
index 669d3891..3298daa7 100644
--- 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
index 8bb8f879..25125618 100644
--- 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