Compilation target architectures 6.0, 6.1, 6.2, & 7.2 do not correspond to any GPUs currently offered by CoreWeave, and bloat the resulting builds, so they are removed as CI defaults. They can still be compiled by manually invoking a Dockerfile build.
Slimmer PyTorch Builds
Compilation target architectures 6.0, 6.1, 6.2, & 7.2 do not correspond to any GPUs currently offered by CoreWeave, and bloat the resulting builds, so they are removed as CI defaults. They can still be compiled by manually invoking a Dockerfile build.
The list of GPU architectures currently offered by CoreWeave are:
H100_NVLINK_80GB
sm_90
H100_PCIE
sm_90
A100_NVLINK_80GB
sm_80
A100_NVLINK
sm_80
A100_PCIE_40GB
sm_80
A100_PCIE_80GB
sm_80
A40
sm_86
RTX_A6000
sm_86
RTX_A5000
sm_86
RTX_A4000
sm_86
Tesla_V100_NVLINK
sm_70
Quadro_RTX_5000
sm_75
Quadro_RTX_4000
sm_75