disable pip cache

vectorch-ai · Mar 1, 2025 · ca34f43 · ca34f43
1 parent daf3dfa
commit ca34f43
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git a/.github/workflows/build_wheel.yml b/.github/workflows/build_wheel.yml
@@ -57,7 +57,6 @@ jobs:
               -e SCALELLM_VERSION_SUFFIX="+cu${CUDA_VERSION//./}torch${TORCH_VERSION}" \
               -e VCPKG_DEFAULT_BINARY_CACHE=/ci_cache/.vcpkg/bincache \
               -e CCACHE_DIR=/ci_cache/.ccache \
-              -e PIP_CACHE_DIR=/ci_cache/.pip \
               -u $(id -u):$(id -g) \
               vectorchai/scalellm_manylinux:cuda${CUDA_VERSION} \
               bash /ScaleLLM/scripts/build_wheel.sh

diff --git a/scripts/build_wheel.sh b/scripts/build_wheel.sh
@@ -27,7 +27,7 @@ export PATH="/opt/python/cp${PYVER}-cp${PYVER}/bin:$PATH"
 python -m pip install --upgrade pip
 
 # install PyTorch
-pip install torch==$TORCH_VERSION --index-url "https://download.pytorch.org/whl/cu${CUDA_VERSION//./}"
+pip install torch==$TORCH_VERSION -i "https://download.pytorch.org/whl/cu${CUDA_VERSION//./}"
 
 # install other dependencies
 pip install numpy jinja2