Drop -links

.github/container/build-jax.sh

@@ -269,8 +269,8 @@ if ! grep 'try-import %workspace%/.local_cuda.bazelrc' "${SRC_PATH_JAX}/.bazelrc
 fi
 cat > "${SRC_PATH_JAX}/.local_cuda.bazelrc" << EOF
 build:cuda --repo_env=LOCAL_CUDA_PATH="/usr/local/cuda"
-build:cuda --repo_env=LOCAL_CUDNN_PATH="/opt/nvidia-links/cudnn"
-build:cuda --repo_env=LOCAL_NCCL_PATH="/opt/nvidia-links/nccl"
+build:cuda --repo_env=LOCAL_CUDNN_PATH="/opt/nvidia/cudnn"
+build:cuda --repo_env=LOCAL_NCCL_PATH="/opt/nvidia/nccl"
 EOF
 time python "${SRC_PATH_JAX}/build/build.py" \
     --editable \

.github/container/install-cudnn.sh

@@ -45,7 +45,7 @@ rm -rf /var/lib/apt/lists/*
 # Create a prefix with include/ and lib/ directories containing symlinks to the cuDNN
 # version that was just installed; this is useful to pass to XLA to avoid it fetching
 # its own copy of cuDNN.
-prefix=/opt/nvidia-links/cudnn
+prefix=/opt/nvidia/cudnn
 if [[ -d "${prefix}" ]]; then
   echo "Skipping link farm creation"
   exit 1

.github/container/install-nccl.sh

@@ -36,7 +36,7 @@ fi
 # Create a prefix with include/ and lib/ directories containing symlinks to the NCCL
 # version installed at the system level; this is useful to pass to XLA to avoid it
 # fetching its own copy.
-prefix=/opt/nvidia-links/nccl
+prefix=/opt/nvidia/nccl
 if [[ -d "${prefix}" ]]; then
   echo "Skipping link farm creation"
   exit 1