adityas129 commited on
Commit
fe2b69d
·
verified ·
1 Parent(s): 7dcdc7e

Update Dockerfile

Browse files
Files changed (1) hide show
  1. 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.7.1" "jaxlib==0.7.1"
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 \