Update python3_devel.yml

jbellister-slac · Feb 26, 2025 · 8aff2e5 · 8aff2e5
1 parent a8036d5
commit 8aff2e5
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/python3_devel.yml b/python3_devel.yml
@@ -1,5 +1,6 @@
 name: python3_devel
 channels:
+  - nvidia/label/cuda-12.3.2
   - conda-forge
 dependencies:
   - alembic
@@ -11,6 +12,7 @@ dependencies:
   - caproto
   - conda-pack
   - cothread
+  - cuda-toolkit
   - cx_oracle
   - dask
   - dateparser
@@ -77,6 +79,7 @@ dependencies:
   - pytest-qt
   - python=3.10
   - python-vxi11
+  - pytorch-gpu=2.6.0
   - pytz
   - pyzmq
   - qtpy