@@ -16,12 +16,10 @@ jobs:
1616 gpu-arch-type : " cuda"
1717 gpu-arch-version : " 12.8"
1818 torch-version : " stable"
19- cmake-version : " 3.22.1"
2019 - runs-on : " linux.g5.12xlarge.nvidia.gpu"
2120 gpu-arch-type : " cuda"
2221 gpu-arch-version : " 12.8"
2322 torch-version : " nightly"
24- cmake-version : " latest"
2523
2624 uses : pytorch/test-infra/.github/workflows/linux_job_v2.yml@main
2725 with :
@@ -31,24 +29,32 @@ jobs:
3129 gpu-arch-version : ${{ matrix.gpu-arch-version }}
3230 script : |
3331 set -ex
32+
33+ # install build tools to system
34+ dnf config-manager --set-enabled powertools
35+ dnf install -y almalinux-release-devel
36+ dnf install -y ninja-build cmake
37+
38+ # Remove old cmake/ninja
39+ rm -f "/opt/conda/bin/ninja" || true
40+ rm -f "/opt/conda/bin/cmake" || true
41+ rm -f "/usr/local/bin/cmake" || true
42+
3443 # use faster libmamba solver
3544 conda config --set solver libmamba
3645
37- conda create -n venv python=3.12 ninja -y
46+ conda create -n venv python=3.12 -y
3847 conda activate venv
3948 python -m pip install --upgrade pip
4049
50+ # Nuke conda libstd++ to avoid conflicts with system toolset
51+ rm -f "$CONDA_PREFIX/lib/libstdc"* || true
52+
4153 # Optionally install torch nightly, pulls latest CUDA from pip otherwise
4254 if [ "${{ matrix.torch-version }}" = "nightly" ]; then
4355 pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu128
4456 fi
4557
46- if [ "${{ matrix.cmake-version }}" = "latest" ]; then
47- conda install -y cmake
48- else # default to latest
49- conda install -y cmake==${{ matrix.cmake-version }}
50- fi
51-
5258 pip install -r requirements.txt
5359
5460 # Build pybindings
0 commit comments