attempt to update docker setup for gpu support

openproblems-bio · Nov 20, 2024 · 5260044 · 5260044
1 parent f1c0f1b
commit 5260044
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/methods/destvi/config.vsh.yaml b/src/methods/destvi/config.vsh.yaml
@@ -35,6 +35,7 @@ engines:
     setup:
       - type: python
         packages: 
+          - jax[cuda12]
           - scvi-tools
 
 runners: