Current versions support with prebuilt wheels:
torch-version: ['2.0.0', '2.1.0', '2.2.0', '2.3.0', '2.4.0']
cuda-version: ['cu118', 'cu121', 'cu124']
Would be great to add current versions of torch (2.5, 2.6, 2.6) and maybe newer CUDA versions (12.6 (default for 2.7), 12.8) - maybe something like:
torch-version: ['2.0.0', '2.1.0', '2.2.0', '2.3.0', '2.4.0', '2.5.0', '2.6.0', '2.7.0']
cuda-version: ['cu118', 'cu124', 'cu126', 'cu128']
Could also cut support for older torch/cuda versions if this is getting to be too many
Current versions support with prebuilt wheels:
Would be great to add current versions of torch (2.5, 2.6, 2.6) and maybe newer CUDA versions (12.6 (default for 2.7), 12.8) - maybe something like:
Could also cut support for older torch/cuda versions if this is getting to be too many