python installation is dizzying

broadinstitute · Jan 10, 2025 · 102a225 · 102a225
1 parent 1b66e04
commit 102a225
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/scripts/gatkcondaenv.yml.template b/scripts/gatkcondaenv.yml.template
@@ -18,6 +18,7 @@ channels:
 # verify that key dependencies are installed from the correct channel
 - conda-forge
 - pytorch
+- nvidia
 
 dependencies:
 
@@ -32,7 +33,7 @@ dependencies:
 - conda-forge::h5py=3.10.0
 - pytorch::pytorch=2.1.0
 - conda-forge::pytorch-lightning=2.4.0   # supports Pytorch >= 2.1 and <= 2.4, used by NVScoreVariants
-- nvidia::pytorch-cuda=12.1
+- pytorch::pytorch-cuda=12.1
 - conda-forge::scikit-learn=1.3.2
 - conda-forge::matplotlib=3.8.2
 - conda-forge::pandas=2.1.3