From 7922898052c40ea9a383a38de2cc0fc08da56853 Mon Sep 17 00:00:00 2001
From: Wei Ji <23487320+weiji14@users.noreply.github.com>
Date: Wed, 13 Dec 2023 11:54:58 +1300
Subject: [PATCH] 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.
---
 ml-notebook/environment.yml      | 2 +-
 pytorch-notebook/environment.yml | 4 ++--
 2 files changed, 3 insertions(+), 3 deletions(-)

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