test

pytorch · Nov 13, 2024 · 08fa4b7 · 08fa4b7
1 parent f2cfe40
commit 08fa4b7
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/.github/scripts/setup-env.sh b/.github/scripts/setup-env.sh
@@ -37,10 +37,6 @@ conda activate ci
 conda install --quiet --yes libjpeg-turbo -c pytorch
 pip install --progress-bar=off --upgrade setuptools==72.1.0
 
-# set ld_library_path - make sure so libraries have precedence
-if [[ "${OS_TYPE}" == linux ]]; then
-  export LD_LIBRARY_PATH="${CONDA_PREFIX}/lib:${LD_LIBRARY_PATH}"
-fi
 
 # See https://github.com/pytorch/vision/issues/6790
 if [[ "${PYTHON_VERSION}" != "3.11" ]]; then

diff --git a/.github/workflows/build-cmake.yml b/.github/workflows/build-cmake.yml
@@ -33,6 +33,11 @@ jobs:
         export PYTHON_VERSION=3.9
         export GPU_ARCH_TYPE=${{ matrix.gpu-arch-type }}
         export GPU_ARCH_VERSION=${{ matrix.gpu-arch-version }}
+        # Prepare conda
+        CONDA_PATH=$(which conda)
+        eval "$(${CONDA_PATH} shell.bash hook)"
+        conda activate ci
+        export LD_LIBRARY_PATH="${CONDA_PREFIX}/lib:${LD_LIBRARY_PATH}"
         ./.github/scripts/cmake.sh
 
   macos: