Unverified Commit e5a1b71d authored by Philip Meier's avatar Philip Meier Committed by GitHub
Browse files

moving windows conda / CUDA workarounds to Nova (#7555)

parent caa02392
...@@ -30,6 +30,8 @@ jobs: ...@@ -30,6 +30,8 @@ jobs:
with: with:
repository: pytorch/vision repository: pytorch/vision
runner: ${{ matrix.runner }} runner: ${{ matrix.runner }}
gpu-arch-type: ${{ matrix.gpu-arch-type }}
gpu-arch-version: ${{ matrix.gpu-arch-version }}
timeout: 120 timeout: 120
script: | script: |
set -euxo pipefail set -euxo pipefail
...@@ -38,14 +40,4 @@ jobs: ...@@ -38,14 +40,4 @@ jobs:
export GPU_ARCH_TYPE=${{ matrix.gpu-arch-type }} export GPU_ARCH_TYPE=${{ matrix.gpu-arch-type }}
export GPU_ARCH_VERSION=${{ matrix.gpu-arch-version }} export GPU_ARCH_VERSION=${{ matrix.gpu-arch-version }}
# TODO: Port this to pytorch/test-infra/.github/workflows/windows_job.yml
export PATH="/c/Jenkins/Miniconda3/Scripts:${PATH}"
if [[ $GPU_ARCH_TYPE == 'cuda' ]]; then
# TODO: This should be handled by the generic Windows job the same as its done by the generic Linux job
export CUDA_HOME="/c/Program Files/NVIDIA GPU Computing Toolkit/CUDA/v${{ matrix.gpu-arch-version }}"
export CUDA_PATH="${CUDA_HOME}"
export PATH="${CUDA_PATH}/bin:${PATH}"
fi
./.github/scripts/unittest.sh ./.github/scripts/unittest.sh
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment