Update Dockerfile
Browse files- Dockerfile +1 -1
Dockerfile
CHANGED
|
@@ -56,7 +56,7 @@ RUN ln -sf /usr/bin/python3.11 /usr/bin/python && python -m pip install --upgrad
|
|
| 56 |
# --- Python dependencies (install order matters!) ---
|
| 57 |
|
| 58 |
# 1) Install JAX first
|
| 59 |
-
RUN python -m pip install "jax[cuda12]==0.
|
| 60 |
|
| 61 |
# 2) Install seqio dependencies manually (excluding tfds-nightly and tensorflow-text)
|
| 62 |
RUN python -m pip install \
|
|
|
|
| 56 |
# --- Python dependencies (install order matters!) ---
|
| 57 |
|
| 58 |
# 1) Install JAX first
|
| 59 |
+
RUN python -m pip install "jax[cuda12]==0.8.1" "jaxlib==0.8.1"
|
| 60 |
|
| 61 |
# 2) Install seqio dependencies manually (excluding tfds-nightly and tensorflow-text)
|
| 62 |
RUN python -m pip install \
|