Commit
·
4130d86
1
Parent(s):
a628b37
claude hallucinated a jax version
Browse files- Dockerfile +1 -1
Dockerfile
CHANGED
|
@@ -58,7 +58,7 @@ RUN ln -sf /usr/bin/python3.11 /usr/bin/python && \
|
|
| 58 |
RUN uv pip install --system tf-nightly
|
| 59 |
|
| 60 |
# Install JAX with CUDA support (pinned for stability)
|
| 61 |
-
RUN uv pip install --system "jax[cuda12]
|
| 62 |
|
| 63 |
# Install base dependencies
|
| 64 |
RUN uv pip install --system \
|
|
|
|
| 58 |
RUN uv pip install --system tf-nightly
|
| 59 |
|
| 60 |
# Install JAX with CUDA support (pinned for stability)
|
| 61 |
+
RUN uv pip install --system "jax[cuda12]" jaxlib
|
| 62 |
|
| 63 |
# Install base dependencies
|
| 64 |
RUN uv pip install --system \
|