normalisation

spaceml-org · May 17, 2024 · 577d4ee · 577d4ee
1 parent 42b440b
commit 577d4ee
Show file tree

Hide file tree

Showing 3 changed files with 519 additions and 4 deletions.
diff --git a/environments/environment_iti.yaml b/environments/environment_iti.yaml
@@ -1,6 +1,8 @@
 name: rs_tools_iti
 channels:
   - conda-forge
+  - pytorch
+  - nvidia
 dependencies:
   - python=3.11
   # Standard Libraries
@@ -24,7 +26,8 @@ dependencies:
   - zarr
   - h5netcdf
   - pyhdf
-  - nvidia/label/cuda-12.2.0::cuda
+  # - nvidia/label/cuda-12.2.0::cuda
+  - pytorch-cuda
   # - h5py
   # GUI
   - ipywidgets