Prebuilt images for xformers with Composable Kernel Flash Attention extensions (AMD MI Instinct GPUs only)
Pytorch: 10/01/2024 nightly ROCm: 6.2.2 Python: 3.11
Content type
Image
Digest
sha256:6b0a9df50…
Size
10.9 GB
Last updated
almost 2 years ago
docker pull rocm/xformers