cached version works now

determined-ai · Dec 3, 2024 · e64d9dd · e64d9dd
1 parent 7324570
commit e64d9dd
Showing 1 changed file with 8 additions and 6 deletions.
diff --git a/.circleci/real_config.yml b/.circleci/real_config.yml
@@ -519,12 +519,14 @@ commands:
                 name: Install Nvidia Apex
                 description: "Apex installation needs Cuda 12.1 because Pytorch binaries were compiled by Cuda 12.1."
                 command: |
-                  wget https://developer.download.nvidia.com/compute/cuda/12.1.0/local_installers/cuda_12.1.0_530.30.02_linux.run
-                  sudo sh cuda_12.1.0_530.30.02_linux.run --toolkit --silent
-                  nvcc --version
-                  git clone https://github.com/NVIDIA/apex ~/apex
-                  git -C ~/apex checkout tags/24.04.01
-                  pip install -v --disable-pip-version-check --no-cache-dir --no-build-isolation --config-settings "--build-option=--cpp_ext" --config-settings "--build-option=--cuda_ext" ~/apex
+                  if ! pip show apex; then
+                    wget https://developer.download.nvidia.com/compute/cuda/12.1.0/local_installers/cuda_12.1.0_530.30.02_linux.run
+                    sudo sh cuda_12.1.0_530.30.02_linux.run --toolkit --silent
+                    nvcc --version
+                    git clone https://github.com/NVIDIA/apex ~/apex
+                    git -C ~/apex checkout tags/24.04.01
+                    pip install -v --disable-pip-version-check --no-cache-dir --no-build-isolation --config-settings "--build-option=--cpp_ext" --config-settings "--build-option=--cuda_ext" ~/apex
+                  fi
 
       - save_cache:
           key: det-python-deps-<<pipeline.parameters.cache-buster>>-{{ checksum "/tmp/cachefile" }}