Skip to content
Snippets Groups Projects
Commit bb0c8724 authored by Michael Thiel's avatar Michael Thiel
Browse files

added jaxlib=cuda

parent 7be2129f
No related branches found
No related tags found
No related merge requests found
Pipeline #233332 failed
Stage: build
Hide whitespace changes
Inline Side-by-side
Showing
with 1 addition and 1 deletion
Dockerfile
+ 1
− 1
View file @ bb0c8724
...@@ -16,7 +16,7 @@ RUN mamba install -c conda-forge myst-nb ...@@ -16,7 +16,7 @@ RUN mamba install -c conda-forge myst-nb
RUN mamba install -c conda-forge ipympl RUN mamba install -c conda-forge ipympl
RUN mamba install -c conda-forge ipywidgets RUN mamba install -c conda-forge ipywidgets
RUN mamba install -c conda-forge jax RUN mamba install -c conda-forge "jaxlib=*=*cuda*" jax
RUN mamba install -c conda-forge keras RUN mamba install -c conda-forge keras
RUN mamba install -c conda-forge control RUN mamba install -c conda-forge control
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment