build_all_linux_wheels.sh 1.01 KB
Newer Older
1
2
#!/bin/bash
# Define the versions for Python, Torch, and CUDA
3
python_versions=("3.10" "3.11" "3.12")
4
5
6
7
8
9
torch_versions=("2.5" "2.6")
cuda_versions=("12.4")

# Loop through all combinations of Python, Torch, and CUDA versions
for python_version in "${python_versions[@]}"; do
  for torch_version in "${torch_versions[@]}"; do
Muyang Li's avatar
Muyang Li committed
10
11
12
13
14
    # Skip building for Python 3.13 and PyTorch 2.5
    if [[ "$python_version" == "3.13" && "$torch_version" == "2.5" ]]; then
      echo "Skipping Python 3.13 with PyTorch 2.5"
      continue
    fi
15
16
17
18
    for cuda_version in "${cuda_versions[@]}"; do
      bash scripts/build_linux_wheel.sh "$python_version" "$torch_version" "$cuda_version"
    done
  done
Muyang Li's avatar
Muyang Li committed
19
20
done

21
22
23
bash scripts/build_linux_wheel.sh "3.10" "2.7" "12.8"
bash scripts/build_linux_wheel.sh "3.11" "2.7" "12.8"
bash scripts/build_linux_wheel.sh "3.12" "2.7" "12.8"
24

25
26
27
#bash scripts/build_linux_wheel_cu128.sh "3.10" "2.8" "12.8"
#bash scripts/build_linux_wheel_cu128.sh "3.11" "2.8" "12.8"
#bash scripts/build_linux_wheel_cu128.sh "3.12" "2.8" "12.8"