Install all wheels to /opt/jaxlib

NVIDIA · Aug 9, 2024 · 7268204 · 7268204
1 parent 046813f
commit 7268204
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/.github/container/build-jax.sh b/.github/container/build-jax.sh
@@ -304,7 +304,7 @@ fi
 
 # after installation (example)
 #  jax                     0.4.32.dev20240808+9c2caedab /opt/jax
-#  jax-cuda12-pjrt         0.4.32.dev20240808           /opt/jaxlibjax_gpu_pjrt
+#  jax-cuda12-pjrt         0.4.32.dev20240808           /opt/jaxlib/jax_gpu_pjrt
 #  jax-cuda12-plugin       0.4.32.dev20240808           /opt/jaxlib/jax_gpu_plugin
 #  jaxlib                  0.4.32.dev20240808           /opt/jaxlib/jaxlib
 pip list | grep jax