Symlink CUDA lib->lib64

NVIDIA · Aug 15, 2024 · a11fec5 · a11fec5
1 parent 8196eae
commit a11fec5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/.github/container/build-jax.sh b/.github/container/build-jax.sh
@@ -256,7 +256,7 @@ popd
 
 ## Build jaxlib
 mkdir -p "${BUILD_PATH_JAXLIB}"
-clean
+ln -s /usr/local/cuda/lib64 /usr/local/cuda/lib
 time python "${SRC_PATH_JAX}/build/build.py" \
     --editable \
     --use_clang \