update pytorch-cuda dependency

abearab · Nov 22, 2024 · fb8f599 · fb8f599
1 parent adf4dc5
commit fb8f599
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/environment.yml b/environment.yml
@@ -7,7 +7,7 @@ channels:
 dependencies:
   - python==3.11
   - pytorch::pytorch
-  - pytorch::pytorch-cuda==12.1
+  - pytorch::pytorch-cuda
   - pytorch::torchvision
   - pytorch::torchaudio
   - conda-forge::torchmetrics