PPU V1.6.0 PIP package manifest

更新时间:
复制 MD 格式

Updates:

  1. Starting with version 1.6.x, OS-free PIP packages are available. You can use the same installation instructions for AI container images on various base operating systems (OS), such as Ubuntu and AliOS. This feature is currently available only when you use the PPU PIP service provided by PTG over the Internet.

  2. Starting with version 1.6.x, the PIP package list is no longer updated using incremental files for minor versions. All updates are consolidated into this single "1.6.x" file. New PIP packages are added weekly. Check this document for the latest version.

  3. The PPU PIP service can be used in two ways. For more information, see Use the PPU PIP service on ACS and Use the PPU PIP source (v1.6.x) in PAI. You can choose one of the following methods:

    1. PPU PIP service provided by PTG: Use the service over the Internet with explicit authentication.

    2. PPU PIP service provided by Alibaba Cloud: Use the service over a VPC without authentication. This is the recommended method.

Note
  • Using the pip install command with the -i IndexURL option may be optional. For more information, see the documentation for your cloud product.

  • The <IndexURL> varies depending on the PPU PIP service that you use. The following are some examples of the <IndexURL>:

    # 1.1 Index URL format for the PPU PIP service provided by PTG
    https://art-pub.eng.t-head.cn/artifactory/api/pypi/ptgai-pypi_ppu_ubuntu_<cuda-version>_index/simple/
    ## Index URL examples for the PPU PIP service provided by PTG:
    https://art-pub.eng.t-head.cn/artifactory/api/pypi/ptgai-pypi_ppu_ubuntu_cu123_index/simple/
    https://art-pub.eng.t-head.cn/artifactory/api/pypi/ptgai-pypi_ppu_ubuntu_cu124_index/simple/
    https://art-pub.eng.t-head.cn/artifactory/api/pypi/ptgai-pypi_ppu_ubuntu_cu126_index/simple/
    https://art-pub.eng.t-head.cn/artifactory/api/pypi/ptgai-pypi_ppu_ubuntu_cu128_index/simple/
    
    # 1.2 Index URL format for the PPU PIP service provided by PTG (OS-free style)
    https://art-pub.eng.t-head.cn/artifactory/api/pypi/<cuda-version>_index/simple/
    ## Index URL examples for the PPU PIP service provided by PTG (OS-free style):
    https://art-pub.eng.t-head.cn/artifactory/api/pypi/cu126_index/simple/
    https://art-pub.eng.t-head.cn/artifactory/api/pypi/cu128_index/simple/
    
    
    # 2.1 Index URL format for the PPU PIP service provided by Alibaba Cloud:
    https://aiext-pypi.mirrors.aliyuncs.com/pg1-pip/ubuntu_<cuda-version>/simple/
    ## Index URL examples for the PPU PIP service provided by Alibaba Cloud:
    https://aiext-pypi.mirrors.aliyuncs.com/pg1-pip/ubuntu_cu123/simple/
    https://aiext-pypi.mirrors.aliyuncs.com/pg1-pip/ubuntu_cu124/simple/
    https://aiext-pypi.mirrors.aliyuncs.com/pg1-pip/ubuntu_cu126/simple/
    https://aiext-pypi.mirrors.aliyuncs.com/pg1-pip/ubuntu_cu128/simple/
    
    ## 2.2 The PPU PIP service provided by Alibaba Cloud does not yet offer an OS-free style Index URL.

Note: This list is updated to version 250902.

Item

Version

OS

CUDA

Framework

Python

Type

Command line (OS-free)

Command line (Ubuntu)

apex

25.4

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.10

ce

pip install apex==25.4 (-i <IndexURL>)

pip install apex==25.4 (-i <IndexURL>)

apex

25.4

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.12

ce

pip install apex==25.4 (-i <IndexURL>)

pip install apex==25.4 (-i <IndexURL>)

apex

25.4

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.8

torch 2.7.0

3.12

ce

pip install apex==25.4 (-i <IndexURL>)

pip install apex==25.4 (-i <IndexURL>)

apex

24.4.1

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.10

ce

pip install apex==24.4.1 (-i <IndexURL>)

pip install apex==24.4.1 (-i <IndexURL>)

apex

24.4.1

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.12

ce

pip install apex==24.4.1 (-i <IndexURL>)

pip install apex==24.4.1 (-i <IndexURL>)

apex

24.4.1

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.8

torch 2.7.0

3.12

ce

pip install apex==24.4.1 (-i <IndexURL>)

pip install apex==24.4.1 (-i <IndexURL>)

apex

25.7

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.10

ce

pip install apex==25.7 (-i <IndexURL>)

pip install apex==25.7 (-i <IndexURL>)

apex

25.7

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.12

ce

pip install apex==25.7 (-i <IndexURL>)

pip install apex==25.7 (-i <IndexURL>)

apex

25.7

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.8

torch 2.7.0

3.12

ce

pip install apex==25.7 (-i <IndexURL>)

pip install apex==25.7 (-i <IndexURL>)

bitsandbytes

0.45.5

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.12

ce

pip install bitsandbytes==0.45.5 (-i <IndexURL>)

pip install bitsandbytes==0.45.5 (-i <IndexURL>)

bitsandbytes

0.45.5

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.8

torch 2.7.0

3.12

ce

pip install bitsandbytes==0.45.5 (-i <IndexURL>)

pip install bitsandbytes==0.45.5 (-i <IndexURL>)

bitsandbytes

0.46.1

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.10

ce

pip install bitsandbytes==0.46.1 (-i <IndexURL>)

pip install bitsandbytes==0.46.1 (-i <IndexURL>)

bitsandbytes

0.46.1

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.8

torch 2.7.0

3.12

ce

pip install bitsandbytes==0.46.1 (-i <IndexURL>)

pip install bitsandbytes==0.46.1 (-i <IndexURL>)

bitsandbytes

0.46.1

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.12

ce

pip install bitsandbytes==0.46.1 (-i <IndexURL>)

pip install bitsandbytes==0.46.1 (-i <IndexURL>)

block_sparse_attn

0.0.1

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.10

ce

pip install block_sparse_attn==0.0.1 (-i <IndexURL>)

pip install block_sparse_attn==0.0.1 (-i <IndexURL>)

block_sparse_attn

0.0.1

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.12

ce

pip install block_sparse_attn==0.0.1 (-i <IndexURL>)

pip install block_sparse_attn==0.0.1 (-i <IndexURL>)

block_sparse_attn

0.0.1

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.8

torch 2.7.0

3.12

ce

pip install block_sparse_attn==0.0.1 (-i <IndexURL>)

pip install block_sparse_attn==0.0.1 (-i <IndexURL>)

byte_flux

1.1.1

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.10

ce

pip install byte_flux==1.1.1 (-i <IndexURL>)

pip install byte_flux==1.1.1 (-i <IndexURL>)

byte_flux

1.1.1

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.12

ce

pip install byte_flux==1.1.1 (-i <IndexURL>)

byte_flux

1.1.1

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.8

torch 2.7.0

3.12

ce

pip install byte_flux==1.1.1 (-i <IndexURL>)

byte_flux

1.0.4

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.10

ce

pip install byte_flux==1.0.4 (-i <IndexURL>)

pip install byte_flux==1.0.4 (-i <IndexURL>)

byte_flux

1.0.4

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.12

ce

pip install byte_flux==1.0.4 (-i <IndexURL>)

byte_flux

1.0.4

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.8

torch 2.7.0

3.12

ce

pip install byte_flux==1.0.4 (-i <IndexURL>)

cpm_kernel

1.0.11

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.10

ce

pip install cpm_kernel==1.0.11 (-i <IndexURL>)

pip install cpm_kernel==1.0.11 (-i <IndexURL>)

cpm_kernel

1.0.11

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.12

ce

pip install cpm_kernel==1.0.11 (-i <IndexURL>)

pip install cpm_kernel==1.0.11 (-i <IndexURL>)

cpm_kernel

1.0.11

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.8

torch 2.7.0

3.12

ce

pip install cpm_kernel==1.0.11 (-i <IndexURL>)

pip install cpm_kernel==1.0.11 (-i <IndexURL>)

cumm-cu126

0.8.2

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

3.10

ce

pip install cumm-cu126==0.8.2 (-i <IndexURL>)

pip install cumm-cu126==0.8.2 (-i <IndexURL>)

cumm-cu126

0.8.2

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

3.12

ce

pip install cumm-cu126==0.8.2 (-i <IndexURL>)

pip install cumm-cu126==0.8.2 (-i <IndexURL>)

cumm-cu128

0.8.2

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.8

3.12

ce

pip install cumm-cu128==0.8.2 (-i <IndexURL>)

pip install cumm-cu128==0.8.2 (-i <IndexURL>)

cumm-cu126

0.8.1

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

3.10

ce

pip install cumm-cu126==0.8.1 (-i <IndexURL>)

pip install cumm-cu126==0.8.1 (-i <IndexURL>)

cumm-cu126

0.8.1

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

3.12

ce

pip install cumm-cu126==0.8.1 (-i <IndexURL>)

pip install cumm-cu126==0.8.1 (-i <IndexURL>)

cumm-cu128

0.8.1

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.8

3.12

ce

pip install cumm-cu128==0.8.1 (-i <IndexURL>)

pip install cumm-cu128==0.8.1 (-i <IndexURL>)

cumm-cu126

0.8.0

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

3.10

ce

pip install cumm-cu126==0.8.0 (-i <IndexURL>)

pip install cumm-cu126==0.8.0 (-i <IndexURL>)

cumm-cu126

0.8.0

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

3.12

ce

pip install cumm-cu126==0.8.0 (-i <IndexURL>)

pip install cumm-cu126==0.8.0 (-i <IndexURL>)

cumm-cu128

0.8.0

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.8

3.12

ce

pip install cumm-cu128==0.8.0 (-i <IndexURL>)

pip install cumm-cu128==0.8.0 (-i <IndexURL>)

cumm-cu126

0.7.13

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

3.10

ce

pip install cumm-cu126==0.7.13 (-i <IndexURL>)

pip install cumm-cu126==0.7.13 (-i <IndexURL>)

cumm-cu126

0.7.13

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

3.12

ce

pip install cumm-cu126==0.7.13 (-i <IndexURL>)

pip install cumm-cu126==0.7.13 (-i <IndexURL>)

cumm-cu128

0.7.13

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.8

3.12

ce

pip install cumm-cu128==0.7.13 (-i <IndexURL>)

pip install cumm-cu128==0.7.13 (-i <IndexURL>)

cupy

13.5.1

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

3.10

ce

pip install cupy==13.5.1 (-i <IndexURL>)

pip install cupy==13.5.1 (-i <IndexURL>)

cupy

13.5.1

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

3.12

ce

pip install cupy==13.5.1 (-i <IndexURL>)

pip install cupy==13.5.1 (-i <IndexURL>)

cupy

13.5.1

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.8

3.12

ce

pip install cupy==13.5.1 (-i <IndexURL>)

pip install cupy==13.5.1 (-i <IndexURL>)

cupy_cuda12x

13.5.1

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

3.10

ce

pip install cupy_cuda12x==13.5.1 (-i <IndexURL>)

pip install cupy_cuda12x==13.5.1 (-i <IndexURL>)

cupy_cuda12x

13.5.1

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

3.12

ce

pip install cupy_cuda12x==13.5.1 (-i <IndexURL>)

pip install cupy_cuda12x==13.5.1 (-i <IndexURL>)

cupy_cuda12x

13.5.1

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.8

3.12

ce

pip install cupy_cuda12x==13.5.1 (-i <IndexURL>)

pip install cupy_cuda12x==13.5.1 (-i <IndexURL>)

cublas_ops

0.1.0

AliOS 7.2

CUDA 12.6

torch 2.6.0

3.10

ce

cublas_ops

0.1.0

Ubuntu 22.04

CUDA 12.6

torch 2.6.0

3.10

ce

pip install cublas_ops==0.1.0 (-i <IndexURL>)

cublas_ops

0.1.0

AliOS 7.2

CUDA 12.6

torch 2.6.0

3.12

ce

cublas_ops

0.1.0

Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.12

ce

pip install cublas_ops==0.1.0 (-i <IndexURL>)

cublas_ops

0.1.0

AliOS 7.2

CUDA 12.8

torch 2.7.0

3.12

ce

cublas_ops

0.1.0

Ubuntu 24.04

CUDA 12.8

torch 2.7.0

3.12

ce

pip install cublas_ops==0.1.0 (-i <IndexURL>)

ctranslate2

4.6.0

AliOS 7.2

CUDA 12.6

torch 2.6.0

3.10

ce

ctranslate2

4.6.0

Ubuntu 22.04

CUDA 12.6

torch 2.6.0

3.10

ce

pip install ctranslate2==4.6.0 (-i <IndexURL>)

causal_conv1d

1.5.2

AliOS 7.2

CUDA 12.6

torch 2.6.0

3.10

ce

causal_conv1d

1.5.2

Ubuntu 22.04

CUDA 12.6

torch 2.6.0

3.10

ce

pip install causal_conv1d==1.5.2 (-i <IndexURL>)

causal_conv1d

1.5.2

AliOS 7.2

CUDA 12.6

torch 2.6.0

3.12

ce

causal_conv1d

1.5.2

Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.12

ce

pip install causal_conv1d==1.5.2 (-i <IndexURL>)

causal_conv1d

1.5.2

AliOS 7.2

CUDA 12.8

torch 2.7.0

3.12

ce

causal_conv1d

1.5.2

Ubuntu 24.04

CUDA 12.8

torch 2.7.0

3.12

ce

pip install causal_conv1d==1.5.2 (-i <IndexURL>)

deepspeed_kernels

0.0.1

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

3.12

ce

pip install deepspeed_kernels==0.0.1 (-i <IndexURL>)

pip install deepspeed_kernels==0.0.1 (-i <IndexURL>)

dgl

2.4.0

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.10

ce

pip install dgl==2.4.0 (-i <IndexURL>)

pip install dgl==2.4.0 (-i <IndexURL>)

dgl

2.4.0

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.12

ce

pip install dgl==2.4.0 (-i <IndexURL>)

pip install dgl==2.4.0 (-i <IndexURL>)

dgl

2.4.0

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.8

torch 2.7.0

3.12

ce

pip install dgl==2.4.0 (-i <IndexURL>)

pip install dgl==2.4.0 (-i <IndexURL>)

easyocr

1.7.2

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.5.1

3.12

ce

pip install easyocr==1.7.2 (-i <IndexURL>)

pip install easyocr==1.7.2 (-i <IndexURL>)

jukebox

0.1.0

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.12

ce

pip install jukebox==0.1.0 (-i <IndexURL>)

pip install jukebox==0.1.0 (-i <IndexURL>)

faiss

1.11.0

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

3.10

ce

pip install faiss==1.11.0 (-i <IndexURL>)

pip install faiss==1.11.0 (-i <IndexURL>)

faiss

1.11.0

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

3.12

ce

pip install faiss==1.11.0 (-i <IndexURL>)

pip install faiss==1.11.0 (-i <IndexURL>)

faiss

1.11.0

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.8

3.12

ce

pip install faiss==1.11.0 (-i <IndexURL>)

pip install faiss==1.11.0 (-i <IndexURL>)

faiss

1.10.0

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

3.10

ce

pip install faiss==1.10.0 (-i <IndexURL>)

pip install faiss==1.10.0 (-i <IndexURL>)

faiss

1.10.0

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

3.12

ce

pip install faiss==1.10.0 (-i <IndexURL>)

pip install faiss==1.10.0 (-i <IndexURL>)

faiss

1.10.0

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.8

3.12

ce

pip install faiss==1.10.0 (-i <IndexURL>)

pip install faiss==1.10.0 (-i <IndexURL>)

gsplat

1.5.3

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.10

ce

pip install gsplat==1.5.3 (-i <IndexURL>)

pip install gsplat==1.5.3 (-i <IndexURL>)

gsplat

1.5.3

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.12

ce

pip install gsplat==1.5.3 (-i <IndexURL>)

pip install gsplat==1.5.3 (-i <IndexURL>)

gsplat

1.5.3

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.8

torch 2.7.0

3.12

ce

pip install gsplat==1.5.3 (-i <IndexURL>)

pip install gsplat==1.5.3 (-i <IndexURL>)

gsplat

1.5.2

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.10

ce

pip install gsplat==1.5.2 (-i <IndexURL>)

pip install gsplat==1.5.2 (-i <IndexURL>)

gsplat

1.5.2

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.12

ce

pip install gsplat==1.5.2 (-i <IndexURL>)

pip install gsplat==1.5.2 (-i <IndexURL>)

gsplat

1.5.2

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.8

torch 2.7.0

3.12

ce

pip install gsplat==1.5.2 (-i <IndexURL>)

pip install gsplat==1.5.2 (-i <IndexURL>)

implicit

0.7.2

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.10

ce

pip install implicit==0.7.2 (-i <IndexURL>)

pip install implicit==0.7.2 (-i <IndexURL>)

implicit

0.7.2

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.12

ce

pip install implicit==0.7.2 (-i <IndexURL>)

pip install implicit==0.7.2 (-i <IndexURL>)

implicit

0.7.2

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.8

torch 2.7.0

3.12

ce

pip install implicit==0.7.2 (-i <IndexURL>)

pip install implicit==0.7.2 (-i <IndexURL>)

implicit

0.7.1

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.10

ce

pip install implicit==0.7.1 (-i <IndexURL>)

pip install implicit==0.7.1 (-i <IndexURL>)

implicit

0.7.1

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.12

ce

pip install implicit==0.7.1 (-i <IndexURL>)

pip install implicit==0.7.1 (-i <IndexURL>)

implicit

0.7.1

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.8

torch 2.7.0

3.12

ce

pip install implicit==0.7.1 (-i <IndexURL>)

pip install implicit==0.7.1 (-i <IndexURL>)

lammps

2024.8.29.0.0

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

3.10

ce

pip install lammps==2024.8.29.0.0 (-i <IndexURL>)

pip install lammps==2024.8.29.0.0 (-i <IndexURL>)

lammps

2024.8.29.0.0

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

3.12

ce

pip install lammps==2024.8.29.0.0 (-i <IndexURL>)

pip install lammps==2024.8.29.0.0 (-i <IndexURL>)

lammps

2024.8.29.0.0

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.8

3.12

ce

pip install lammps==2024.8.29.0.0 (-i <IndexURL>)

pip install lammps==2024.8.29.0.0 (-i <IndexURL>)

lightgbm

4.6.0

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

3.10

ce

pip install lightgbm==4.6.0 (-i <IndexURL>)

pip install lightgbm==4.6.0 (-i <IndexURL>)

lightgbm

4.6.0

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

3.12

ce

pip install lightgbm==4.6.0 (-i <IndexURL>)

pip install lightgbm==4.6.0 (-i <IndexURL>)

lightgbm

4.6.0

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.8

3.12

ce

pip install lightgbm==4.6.0 (-i <IndexURL>)

pip install lightgbm==4.6.0 (-i <IndexURL>)

lightgbm

4.5.0

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

3.10

ce

pip install lightgbm==4.5.0 (-i <IndexURL>)

pip install lightgbm==4.5.0 (-i <IndexURL>)

lightgbm

4.5.0

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

3.12

ce

pip install lightgbm==4.5.0 (-i <IndexURL>)

pip install lightgbm==4.5.0 (-i <IndexURL>)

lightgbm

4.5.0

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.8

3.12

ce

pip install lightgbm==4.5.0 (-i <IndexURL>)

pip install lightgbm==4.5.0 (-i <IndexURL>)

lightllm

1.0.1

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.10

ce

pip install lightllm==1.0.1 (-i <IndexURL>)

pip install lightllm==1.0.1 (-i <IndexURL>)

lightllm

1.0.1

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.12

ce

pip install lightllm==1.0.1 (-i <IndexURL>)

pip install lightllm==1.0.1 (-i <IndexURL>)

lightllm

1.0.1

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.8

torch 2.7.0

3.12

ce

pip install lightllm==1.0.1 (-i <IndexURL>)

pip install lightllm==1.0.1 (-i <IndexURL>)

lightllm

1.0.0

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.10

ce

pip install lightllm==1.0.0 (-i <IndexURL>)

pip install lightllm==1.0.0 (-i <IndexURL>)

lightllm

1.0.0

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.12

ce

pip install lightllm==1.0.0 (-i <IndexURL>)

pip install lightllm==1.0.0 (-i <IndexURL>)

lightllm

1.0.0

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.8

torch 2.7.0

3.12

ce

pip install lightllm==1.0.0 (-i <IndexURL>)

pip install lightllm==1.0.0 (-i <IndexURL>)

lmdeploy

0.8.0

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.10

ce

pip install lmdeploy==0.8.0 (-i <IndexURL>)

pip install lmdeploy==0.8.0 (-i <IndexURL>)

lmdeploy

0.8.0

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.12

ce

pip install lmdeploy==0.8.0 (-i <IndexURL>)

pip install lmdeploy==0.8.0 (-i <IndexURL>)

lmdeploy

0.9.1

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.10

ce

pip install lmdeploy==0.9.1 (-i <IndexURL>)

pip install lmdeploy==0.9.1 (-i <IndexURL>)

lmdeploy

0.9.1

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.12

ce

pip install lmdeploy==0.9.1 (-i <IndexURL>)

pip install lmdeploy==0.9.1 (-i <IndexURL>)

lmdeploy

0.9.1

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.8

torch 2.7.0

3.12

ce

pip install lmdeploy==0.9.1 (-i <IndexURL>)

pip install lmdeploy==0.9.1 (-i <IndexURL>)

rmm_cu12

25.04.00

AliOS 7.2

CUDA 12.6

3.10

ce

rmm_cu12

25.04.00

Ubuntu 22.04

CUDA 12.6

3.10

ce

pip install rmm_cu12==25.04.00 (-i <IndexURL>)

rmm_cu12

25.04.00

AliOS 7.2

CUDA 12.6

3.12

ce

rmm_cu12

25.04.00

Ubuntu 24.04

CUDA 12.6

3.12

ce

pip install rmm_cu12==25.04.00 (-i <IndexURL>)

rmm_cu12

25.04.00

AliOS 7.2

CUDA 12.8

3.12

ce

rmm_cu12

25.04.00

Ubuntu 24.04

CUDA 12.8

3.12

ce

pip install rmm_cu12==25.04.00 (-i <IndexURL>)

rmm_cu12

25.06.00

AliOS 7.2

CUDA 12.6

3.10

ce

rmm_cu12

25.06.00

Ubuntu 22.04

CUDA 12.6

3.10

ce

pip install rmm_cu12==25.06.00 (-i <IndexURL>)

rmm_cu12

25.06.00

AliOS 7.2

CUDA 12.6

3.12

ce

rmm_cu12

25.06.00

Ubuntu 24.04

CUDA 12.6

3.12

ce

pip install rmm_cu12==25.06.00 (-i <IndexURL>)

rmm_cu12

25.06.00

Ubuntu 24.04

CUDA 12.8

3.12

ce

pip install rmm_cu12==25.06.00 (-i <IndexURL>)

mamba_ssm

2.2.3

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.10

ce

pip install mamba_ssm==2.2.3 (-i <IndexURL>)

pip install mamba_ssm==2.2.3 (-i <IndexURL>)

mamba_ssm

2.2.3

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.12

ce

pip install mamba_ssm==2.2.3 (-i <IndexURL>)

pip install mamba_ssm==2.2.3 (-i <IndexURL>)

mamba_ssm

2.2.3

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.8

torch 2.7.0

3.12

ce

pip install mamba_ssm==2.2.3 (-i <IndexURL>)

pip install mamba_ssm==2.2.3 (-i <IndexURL>)

mamba_ssm

2.2.4

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.10

ce

pip install mamba_ssm==2.2.4 (-i <IndexURL>)

pip install mamba_ssm==2.2.4 (-i <IndexURL>)

mamba_ssm

2.2.4

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.12

ce

pip install mamba_ssm==2.2.4 (-i <IndexURL>)

pip install mamba_ssm==2.2.4 (-i <IndexURL>)

mamba_ssm

2.2.4

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.8

torch 2.7.0

3.12

ce

pip install mamba_ssm==2.2.4 (-i <IndexURL>)

pip install mamba_ssm==2.2.4 (-i <IndexURL>)

megablocks

0.9.0

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.10

ce

pip install megablocks==0.9.0 (-i <IndexURL>)

pip install megablocks==0.9.0 (-i <IndexURL>)

megablocks

0.9.0

Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.12

ce

pip install megablocks==0.9.0 (-i <IndexURL>)

megablocks

0.9.0

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.8

torch 2.7.0

3.12

ce

pip install megablocks==0.9.0 (-i <IndexURL>)

pip install megablocks==0.9.0 (-i <IndexURL>)

megablocks

0.10.0

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.10

ce

pip install megablocks==0.10.0 (-i <IndexURL>)

pip install megablocks==0.10.0 (-i <IndexURL>)

megablocks

0.10.0

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.12

ce

pip install megablocks==0.10.0 (-i <IndexURL>)

pip install megablocks==0.10.0 (-i <IndexURL>)

megablocks

0.10.0

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.8

torch 2.7.0

3.12

ce

pip install megablocks==0.10.0 (-i <IndexURL>)

pip install megablocks==0.10.0 (-i <IndexURL>)

mmcv

2.2.0

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.10

ce

pip install mmcv==2.2.0 (-i <IndexURL>)

pip install mmcv==2.2.0 (-i <IndexURL>)

mmcv

2.2.0

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.12

ce

pip install mmcv==2.2.0 (-i <IndexURL>)

pip install mmcv==2.2.0 (-i <IndexURL>)

mmcv

2.2.0

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.8

torch 2.7.0

3.12

ce

pip install mmcv==2.2.0 (-i <IndexURL>)

pip install mmcv==2.2.0 (-i <IndexURL>)

mmcv

2.1.0

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.10

ce

pip install mmcv==2.1.0 (-i <IndexURL>)

pip install mmcv==2.1.0 (-i <IndexURL>)

mmcv

2.1.0

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.12

ce

pip install mmcv==2.1.0 (-i <IndexURL>)

pip install mmcv==2.1.0 (-i <IndexURL>)

mmcv

2.1.0

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.8

torch 2.7.0

3.12

ce

pip install mmcv==2.1.0 (-i <IndexURL>)

pip install mmcv==2.1.0 (-i <IndexURL>)

mmcv-full

1.7.2

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.10

ce

pip install mmcv-full==1.7.2 (-i <IndexURL>)

pip install mmcv-full==1.7.2 (-i <IndexURL>)

mmcv-full

1.7.2

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.12

ce

pip install mmcv-full==1.7.2 (-i <IndexURL>)

pip install mmcv-full==1.7.2 (-i <IndexURL>)

mmcv-full

1.7.2

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.8

torch 2.7.0

3.12

ce

pip install mmcv-full==1.7.2 (-i <IndexURL>)

pip install mmcv-full==1.7.2 (-i <IndexURL>)

mmdet3d

1.4.0

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.10

ce

pip install mmdet3d==1.4.0 (-i <IndexURL>)

pip install mmdet3d==1.4.0 (-i <IndexURL>)

mmdet3d

1.4.0

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.12

ce

pip install mmdet3d==1.4.0 (-i <IndexURL>)

pip install mmdet3d==1.4.0 (-i <IndexURL>)

mmdet3d

1.4.0

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.8

torch 2.7.0

3.12

ce

pip install mmdet3d==1.4.0 (-i <IndexURL>)

pip install mmdet3d==1.4.0 (-i <IndexURL>)

mmdet3d

1.3.0

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.10

ce

pip install mmdet3d==1.3.0 (-i <IndexURL>)

pip install mmdet3d==1.3.0 (-i <IndexURL>)

mmdet3d

1.3.0

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.12

ce

pip install mmdet3d==1.3.0 (-i <IndexURL>)

pip install mmdet3d==1.3.0 (-i <IndexURL>)

mmdet3d

1.3.0

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.8

torch 2.7.0

3.12

ce

pip install mmdet3d==1.3.0 (-i <IndexURL>)

pip install mmdet3d==1.3.0 (-i <IndexURL>)

mmpretrain

1.2.0

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

3.10

ce

pip install mmpretrain==1.2.0 (-i <IndexURL>)

pip install mmpretrain==1.2.0 (-i <IndexURL>)

mmpretrain

1.2.0

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

3.12

ce

pip install mmpretrain==1.2.0 (-i <IndexURL>)

pip install mmpretrain==1.2.0 (-i <IndexURL>)

mmpretrain

1.2.0

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.8

3.12

ce

pip install mmpretrain==1.2.0 (-i <IndexURL>)

pip install mmpretrain==1.2.0 (-i <IndexURL>)

mmsegmentation

1.2.2

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.10

ce

pip install mmsegmentation==1.2.2 (-i <IndexURL>)

pip install mmsegmentation==1.2.2 (-i <IndexURL>)

mmsegmentation

1.2.2

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.12

ce

pip install mmsegmentation==1.2.2 (-i <IndexURL>)

pip install mmsegmentation==1.2.2 (-i <IndexURL>)

mmsegmentation

1.2.2

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.8

torch 2.7.0

3.12

ce

pip install mmsegmentation==1.2.2 (-i <IndexURL>)

pip install mmsegmentation==1.2.2 (-i <IndexURL>)

MinkowskiEngine

0.5.4

AliOS 7.2

CUDA 12.6

torch 2.6.0

3.10

ce

MinkowskiEngine

0.5.4

Ubuntu 22.04

CUDA 12.6

torch 2.6.0

3.10

ce

pip install MinkowskiEngine==0.5.4 (-i <IndexURL>)

MinkowskiEngine

0.5.4

AliOS 7.2

CUDA 12.6

torch 2.6.0

3.12

ce

MinkowskiEngine

0.5.4

Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.12

ce

pip install MinkowskiEngine==0.5.4 (-i <IndexURL>)

MinkowskiEngine

0.5.4

AliOS 7.2

CUDA 12.8

torch 2.7.0

3.12

ce

MinkowskiEngine

0.5.4

Ubuntu 24.04

CUDA 12.8

torch 2.7.0

3.12

ce

pip install MinkowskiEngine==0.5.4 (-i <IndexURL>)

natten

0.17.5

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.10

ce

pip install natten==0.17.5 (-i <IndexURL>)

pip install natten==0.17.5 (-i <IndexURL>)

natten

0.17.5

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.12

ce

pip install natten==0.17.5 (-i <IndexURL>)

pip install natten==0.17.5 (-i <IndexURL>)

natten

0.17.5

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.8

torch 2.7.0

3.12

ce

pip install natten==0.17.5 (-i <IndexURL>)

pip install natten==0.17.5 (-i <IndexURL>)

numba

0.61.2

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

3.10

ce

pip install numba==0.61.2 (-i <IndexURL>)

pip install numba==0.61.2 (-i <IndexURL>)

numba

0.61.2

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

3.12

ce

pip install numba==0.61.2 (-i <IndexURL>)

pip install numba==0.61.2 (-i <IndexURL>)

numba

0.61.2

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.8

3.12

ce

pip install numba==0.61.2 (-i <IndexURL>)

pip install numba==0.61.2 (-i <IndexURL>)

numba

0.61.1

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

3.10

ce

pip install numba==0.61.1 (-i <IndexURL>)

pip install numba==0.61.1 (-i <IndexURL>)

numba

0.61.1

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

3.12

ce

pip install numba==0.61.1 (-i <IndexURL>)

pip install numba==0.61.1 (-i <IndexURL>)

numba

0.61.1

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.8

3.12

ce

pip install numba==0.61.1 (-i <IndexURL>)

pip install numba==0.61.1 (-i <IndexURL>)

nvdiffrast

0.3.3

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.10

ce

pip install nvdiffrast==0.3.3 (-i <IndexURL>)

pip install nvdiffrast==0.3.3 (-i <IndexURL>)

nvdiffrast

0.3.3

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.12

ce

pip install nvdiffrast==0.3.3 (-i <IndexURL>)

pip install nvdiffrast==0.3.3 (-i <IndexURL>)

nvdiffrast

0.3.3

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.8

torch 2.7.0

3.12

ce

pip install nvdiffrast==0.3.3 (-i <IndexURL>)

pip install nvdiffrast==0.3.3 (-i <IndexURL>)

nvidia_dali_cuda120

1.44.0

AliOS 7.2

CUDA 12.6

3.10

ce

nvidia_dali_cuda120

1.44.0

Ubuntu 24.04

CUDA 12.6

3.12

ce

pip install nvidia_dali_cuda120==1.44.0 (-i <IndexURL>)

nvidia_dali_cuda120

1.44.0

AliOS 7.2

CUDA 12.6

3.12

ce

nvidia_dali_cuda120

1.44.0

AliOS 7.2

CUDA 12.8

3.12

ce

nvidia_dali_cuda120

1.44.0

Ubuntu 24.04

CUDA 12.8

3.12

ce

pip install nvidia_dali_cuda120==1.44.0 (-i <IndexURL>)

nvidia_dali_cuda120

1.50.0

AliOS 7.2

CUDA 12.6

3.10

ce

nvidia_dali_cuda120

1.50.0

Ubuntu 22.04

CUDA 12.6

3.10

ce

pip install nvidia_dali_cuda120==1.50.0 (-i <IndexURL>)

nvidia_dali_cuda120

1.50.0

AliOS 7.2

CUDA 12.6

3.12

ce

nvidia_dali_cuda120

1.50.0

Ubuntu 24.04

CUDA 12.6

3.12

ce

pip install nvidia_dali_cuda120==1.50.0 (-i <IndexURL>)

nvidia_dali_cuda120

1.50.0

AliOS 7.2

CUDA 12.8

3.12

ce

nvidia_dali_cuda120

1.50.0

Ubuntu 24.04

CUDA 12.8

3.12

ce

pip install nvidia_dali_cuda120==1.50.0 (-i <IndexURL>)

opencv_contrib_python

4.10.0.84

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

3.10

ce

pip install opencv_contrib_python==4.10.0.84 (-i <IndexURL>)

pip install opencv_contrib_python==4.10.0.84 (-i <IndexURL>)

opencv_contrib_python

4.10.0.84

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

3.12

ce

pip install opencv_contrib_python==4.10.0.84 (-i <IndexURL>)

pip install opencv_contrib_python==4.10.0.84 (-i <IndexURL>)

opencv_contrib_python

4.10.0.84

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.8

3.12

ce

pip install opencv_contrib_python==4.10.0.84 (-i <IndexURL>)

pip install opencv_contrib_python==4.10.0.84 (-i <IndexURL>)

opencv_contrib_python

4.10.0.88

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

3.10

ce

pip install opencv_contrib_python==4.10.0.88 (-i <IndexURL>)

pip install opencv_contrib_python==4.10.0.88 (-i <IndexURL>)

opencv_contrib_python

4.10.0.88

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

3.12

ce

pip install opencv_contrib_python==4.10.0.88 (-i <IndexURL>)

pip install opencv_contrib_python==4.10.0.88 (-i <IndexURL>)

opencv_contrib_python

4.10.0.88

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.8

3.12

ce

pip install opencv_contrib_python==4.10.0.88 (-i <IndexURL>)

pip install opencv_contrib_python==4.10.0.88 (-i <IndexURL>)

pytorch3d

0.7.8

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.10

ce

pip install pytorch3d==0.7.8 (-i <IndexURL>)

pip install pytorch3d==0.7.8 (-i <IndexURL>)

pytorch3d

0.7.8

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.12

ce

pip install pytorch3d==0.7.8 (-i <IndexURL>)

pip install pytorch3d==0.7.8 (-i <IndexURL>)

pytorch3d

0.7.8

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.8

torch 2.7.0

3.12

ce

pip install pytorch3d==0.7.8 (-i <IndexURL>)

pip install pytorch3d==0.7.8 (-i <IndexURL>)

pytorch3d

0.7.7

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.10

ce

pip install pytorch3d==0.7.7 (-i <IndexURL>)

pip install pytorch3d==0.7.7 (-i <IndexURL>)

pytorch3d

0.7.7

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.12

ce

pip install pytorch3d==0.7.7 (-i <IndexURL>)

pip install pytorch3d==0.7.7 (-i <IndexURL>)

pytorch3d

0.7.7

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.8

torch 2.7.0

3.12

ce

pip install pytorch3d==0.7.7 (-i <IndexURL>)

pip install pytorch3d==0.7.7 (-i <IndexURL>)

sageattention

2.0.1

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.10

ce

pip install sageattention==2.0.1 (-i <IndexURL>)

pip install sageattention==2.0.1 (-i <IndexURL>)

sageattention

2.0.1

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.12

ce

pip install sageattention==2.0.1 (-i <IndexURL>)

pip install sageattention==2.0.1 (-i <IndexURL>)

sageattention

2.0.1

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.8

torch 2.7.0

3.12

ce

pip install sageattention==2.0.1 (-i <IndexURL>)

pip install sageattention==2.0.1 (-i <IndexURL>)

sageattention

1.0.6

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.10

ce

pip install sageattention==1.0.6 (-i <IndexURL>)

pip install sageattention==1.0.6 (-i <IndexURL>)

sageattention

1.0.6

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.12

ce

pip install sageattention==1.0.6 (-i <IndexURL>)

pip install sageattention==1.0.6 (-i <IndexURL>)

sageattention

1.0.6

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.8

torch 2.7.0

3.12

ce

pip install sageattention==1.0.6 (-i <IndexURL>)

pip install sageattention==1.0.6 (-i <IndexURL>)

sageattention

1.0.5

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.10

ce

pip install sageattention==1.0.5 (-i <IndexURL>)

pip install sageattention==1.0.5 (-i <IndexURL>)

sageattention

1.0.5

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.12

ce

pip install sageattention==1.0.5 (-i <IndexURL>)

pip install sageattention==1.0.5 (-i <IndexURL>)

sageattention

1.0.5

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.8

torch 2.7.0

3.12

ce

pip install sageattention==1.0.5 (-i <IndexURL>)

pip install sageattention==1.0.5 (-i <IndexURL>)

SwissArmyTransformer

0.4.12

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.10

ce

pip install SwissArmyTransformer==0.4.12 (-i <IndexURL>)

pip install SwissArmyTransformer==0.4.12 (-i <IndexURL>)

SwissArmyTransformer

0.4.12

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.12

ce

pip install SwissArmyTransformer==0.4.12 (-i <IndexURL>)

pip install SwissArmyTransformer==0.4.12 (-i <IndexURL>)

SwissArmyTransformer

0.4.12

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.8

torch 2.7.0

3.12

ce

pip install SwissArmyTransformer==0.4.12 (-i <IndexURL>)

pip install SwissArmyTransformer==0.4.12 (-i <IndexURL>)

spconv-cu126

2.3.6

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

3.10

ce

pip install spconv-cu126==2.3.6 (-i <IndexURL>)

pip install spconv-cu126==2.3.6 (-i <IndexURL>)

spconv-cu126

2.3.6

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

3.12

ce

pip install spconv-cu126==2.3.6 (-i <IndexURL>)

pip install spconv-cu126==2.3.6 (-i <IndexURL>)

spconv-cu126

2.3.6

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.8

3.12

ce

pip install spconv-cu128==2.3.6 (-i <IndexURL>)

pip install spconv-cu128==2.3.6 (-i <IndexURL>)

spconv-cu126

2.3.8

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

3.10

ce

pip install spconv-cu126==2.3.6 (-i <IndexURL>)

pip install spconv-cu126==2.3.6 (-i <IndexURL>)

spconv-cu126

2.3.8

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

3.12

ce

pip install spconv-cu126==2.3.6 (-i <IndexURL>)

pip install spconv-cu126==2.3.6 (-i <IndexURL>)

spconv-cu126

2.3.8

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.8

3.12

ce

pip install spconv-cu128==2.3.6 (-i <IndexURL>)

pip install spconv-cu128==2.3.6 (-i <IndexURL>)

tilelang

0.1.5

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.10

ce

pip install tilelang==0.1.5 (-i <IndexURL>)

pip install tilelang==0.1.5 (-i <IndexURL>)

tilelang

0.1.5

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.12

ce

pip install tilelang==0.1.5 (-i <IndexURL>)

pip install tilelang==0.1.5 (-i <IndexURL>)

tilelang

0.1.5

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.8

torch 2.7.0

3.12

ce

pip install tilelang==0.1.5 (-i <IndexURL>)

pip install tilelang==0.1.5 (-i <IndexURL>)

open3d

0.19.0

Ubuntu 22.04

CUDA 12.6

torch 2.6.0

3.10

ce

pip install open3d==0.19.0 (-i <IndexURL>)

open3d

0.19.0

Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.12

ce

pip install open3d==0.19.0 (-i <IndexURL>)

open3d

0.19.0

Ubuntu 24.04

CUDA 12.8

torch 2.7.0

3.12

ce

pip install open3d==0.19.0 (-i <IndexURL>)

paddlepaddle_gpu

3.0.0

Ubuntu 22.04

CUDA 12.6

3.10

ce

pip install paddlepaddle_gpu==3.0.0 (-i <IndexURL>)

paddlepaddle_gpu

3.0.0

Ubuntu 24.04

CUDA 12.6

3.12

ce

pip install paddlepaddle_gpu==3.0.0 (-i <IndexURL>)

paddlepaddle_gpu

3.0.0

Ubuntu 24.04

CUDA 12.8

3.12

ce

pip install paddlepaddle_gpu==3.0.0 (-i <IndexURL>)

pynvcodec

0.1.0

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

3.10

ce

pip install pynvcodec==0.1.0 (-i <IndexURL>)

pip install pynvcodec==0.1.0 (-i <IndexURL>)

pynvcodec

0.1.0

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

3.12

ce

pip install pynvcodec==0.1.0 (-i <IndexURL>)

pip install pynvcodec==0.1.0 (-i <IndexURL>)

pynvcodec

0.1.0

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.8

3.12

ce

pip install pynvcodec==0.1.0 (-i <IndexURL>)

pip install pynvcodec==0.1.0 (-i <IndexURL>)

pytorch_sparse

0.6.18

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.10

ce

pip install pytorch_sparse==0.6.18 (-i <IndexURL>)

pip install pytorch_sparse==0.6.18 (-i <IndexURL>)

pytorch_sparse

0.6.18

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.12

ce

pip install pytorch_sparse==0.6.18 (-i <IndexURL>)

pip install pytorch_sparse==0.6.18 (-i <IndexURL>)

pytorch_sparse

0.6.18

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.8

torch 2.7.0

3.12

ce

pip install pytorch_sparse==0.6.18 (-i <IndexURL>)

pip install pytorch_sparse==0.6.18 (-i <IndexURL>)

taichi

1.7.2

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

3.10

ce

pip install taichi==1.7.2 (-i <IndexURL>)

pip install taichi==1.7.2 (-i <IndexURL>)

taichi

1.7.2

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

3.12

ce

pip install taichi==1.7.2 (-i <IndexURL>)

pip install taichi==1.7.2 (-i <IndexURL>)

taichi

1.7.2

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.8

3.12

ce

pip install taichi==1.7.2 (-i <IndexURL>)

pip install taichi==1.7.2 (-i <IndexURL>)

taichi

1.7.3

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

3.10

ce

pip install taichi==1.7.3 (-i <IndexURL>)

pip install taichi==1.7.3 (-i <IndexURL>)

taichi

1.7.3

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

3.12

ce

pip install taichi==1.7.3 (-i <IndexURL>)

pip install taichi==1.7.3 (-i <IndexURL>)

taichi

1.7.3

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.8

3.12

ce

pip install taichi==1.7.3 (-i <IndexURL>)

pip install taichi==1.7.3 (-i <IndexURL>)

tilefusion

1.0.0

Ubuntu 24.04

CUDA 12.8

3.12

ce

pip install tilefusion==1.0.0 (-i <IndexURL>)

torchcodec

0.4.0

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.10

ce

pip install torchcodec==0.4.0 (-i <IndexURL>)

pip install torchcodec==0.4.0 (-i <IndexURL>)

torchcodec

0.4.0

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.12

ce

pip install torchcodec==0.4.0 (-i <IndexURL>)

pip install torchcodec==0.4.0 (-i <IndexURL>)

torchcodec

0.4.0

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.8

torch 2.7.0

3.12

ce

pip install torchcodec==0.4.0 (-i <IndexURL>)

pip install torchcodec==0.4.0 (-i <IndexURL>)

torch_scatter

2.1.2

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.10

ce

pip install torch_scatter==2.1.2 (-i <IndexURL>)

pip install torch_scatter==2.1.2 (-i <IndexURL>)

torch_scatter

2.1.2

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.12

ce

pip install torch_scatter==2.1.2 (-i <IndexURL>)

pip install torch_scatter==2.1.2 (-i <IndexURL>)

torch_scatter

2.1.2

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.8

torch 2.7.0

3.12

ce

pip install torch_scatter==2.1.2 (-i <IndexURL>)

pip install torch_scatter==2.1.2 (-i <IndexURL>)

torch_scatter

2.1.1

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.10

ce

pip install torch_scatter==2.1.1 (-i <IndexURL>)

pip install torch_scatter==2.1.1 (-i <IndexURL>)

torch_scatter

2.1.1

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.12

ce

pip install torch_scatter==2.1.1 (-i <IndexURL>)

pip install torch_scatter==2.1.1 (-i <IndexURL>)

torch_scatter

2.1.1

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.8

torch 2.7.0

3.12

ce

pip install torch_scatter==2.1.1 (-i <IndexURL>)

pip install torch_scatter==2.1.1 (-i <IndexURL>)

torchani

2.2.4

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.10

ce

pip install torchani==2.2.4 (-i <IndexURL>)

pip install torchani==2.2.4 (-i <IndexURL>)

torchani

2.2.4

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.12

ce

pip install torchani==2.2.4 (-i <IndexURL>)

pip install torchani==2.2.4 (-i <IndexURL>)

torchani

2.2.4

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.8

torch 2.7.0

3.12

ce

pip install torchani==2.2.4 (-i <IndexURL>)

pip install torchani==2.2.4 (-i <IndexURL>)

torchrl

0.9.1

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.10

ce

pip install torchrl==0.9.1 (-i <IndexURL>)

pip install torchrl==0.9.1 (-i <IndexURL>)

torchrl

0.9.1

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.12

ce

pip install torchrl==0.9.1 (-i <IndexURL>)

pip install torchrl==0.9.1 (-i <IndexURL>)

torchrl

0.9.1

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.8

torch 2.7.0

3.12

ce

pip install torchrl==0.9.1 (-i <IndexURL>)

pip install torchrl==0.9.1 (-i <IndexURL>)

torchrl

0.8.1

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.10

ce

pip install torchrl==0.8.1 (-i <IndexURL>)

pip install torchrl==0.8.1 (-i <IndexURL>)

torchrl

0.8.1

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.12

ce

pip install torchrl==0.8.1 (-i <IndexURL>)

pip install torchrl==0.8.1 (-i <IndexURL>)

torchrl

0.8.1

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.8

torch 2.7.0

3.12

ce

pip install torchrl==0.8.1 (-i <IndexURL>)

pip install torchrl==0.8.1 (-i <IndexURL>)

warp

1.8.0

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.10

ce

pip install warp==1.8.0 (-i <IndexURL>)

pip install warp==1.8.0 (-i <IndexURL>)

warp

1.8.0

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.12

ce

pip install warp==1.8.0 (-i <IndexURL>)

pip install warp==1.8.0 (-i <IndexURL>)

warp

1.8.0

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.8

torch 2.7.0

3.12

ce

pip install warp==1.8.0 (-i <IndexURL>)

pip install warp==1.8.0 (-i <IndexURL>)

torchaudio

2.7.0

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.8

torch 2.7.0

3.12

ce

pip install torchaudio==2.7.0 (-i <IndexURL>)

pip install torchaudio==2.7.0 (-i <IndexURL>)

torchaudio

2.6.0

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.10

ce

pip install torchaudio==2.6.0 (-i <IndexURL>)

pip install torchaudio==2.6.0 (-i <IndexURL>)

torchaudio

2.6.0

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.12

ce

pip install torchaudio==2.6.0 (-i <IndexURL>)

pip install torchaudio==2.6.0 (-i <IndexURL>)

torchaudio

2.5.1

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.5.1

3.10

ce

pip install torchaudio==2.5.1 (-i <IndexURL>)

pip install torchaudio==2.5.1 (-i <IndexURL>)

torchtext

0.18.0

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.10

ce

pip install torchtext==0.18.0 (-i <IndexURL>)

pip install torchtext==0.18.0 (-i <IndexURL>)

torchtext

0.18.0

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.12

ce

pip install torchtext==0.18.0 (-i <IndexURL>)

pip install torchtext==0.18.0 (-i <IndexURL>)

torchdata

0.11.0

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.10

ce

pip install torchdata==0.11.0 (-i <IndexURL>)

pip install torchdata==0.11.0 (-i <IndexURL>)

torchdata

0.11.0

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.12

ce

pip install torchdata==0.11.0 (-i <IndexURL>)

pip install torchdata==0.11.0 (-i <IndexURL>)

torchdata

0.11.0

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.8

torch 2.7.0

3.12

ce

pip install torchdata==0.11.0 (-i <IndexURL>)

pip install torchdata==0.11.0 (-i <IndexURL>)

torchdata

0.10.1

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.5.1

3.10

ce

pip install torchdata==0.10.1 (-i <IndexURL>)

pip install torchdata==0.10.1 (-i <IndexURL>)

torchdata

0.10.1

Ubuntu 24.04

CUDA 12.6

torch 2.5.1

3.12

ce

pip install torchdata==0.10.1 (-i <IndexURL>)

torchvision

0.21.0

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.10

ce

pip install torchvision==0.21.0 (-i <IndexURL>)

pip install torchvision==0.21.0 (-i <IndexURL>)

torchvision

0.21.0

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.12

ce

pip install torchvision==0.21.0 (-i <IndexURL>)

pip install torchvision==0.21.0 (-i <IndexURL>)

torchvision

0.22.0

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.8

torch 2.7.0

3.12

ce

pip install torchvision==0.22.0 (-i <IndexURL>)

pip install torchvision==0.22.0 (-i <IndexURL>)

torchvision

0.20.1

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.5.1

3.10

ce

pip install torchvision==0.20.1 (-i <IndexURL>)

pip install torchvision==0.20.1 (-i <IndexURL>)

deep_ep

1.0.0

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.10

oe

pip install deep_ep==1.0.0 (-i <IndexURL>)

pip install deep_ep==1.0.0 (-i <IndexURL>)

deep_ep

1.0.0

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.12

oe

pip install deep_ep==1.0.0 (-i <IndexURL>)

pip install deep_ep==1.0.0 (-i <IndexURL>)

deep_ep

1.0.0

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.8

torch 2.7.0

3.12

oe

pip install deep_ep==1.0.0 (-i <IndexURL>)

pip install deep_ep==1.0.0 (-i <IndexURL>)

deep_gemm

1.0.0

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.10

oe

pip install deep_gemm==1.0.0 (-i <IndexURL>)

pip install deep_gemm==1.0.0 (-i <IndexURL>)

deep_gemm

1.0.0

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.12

oe

pip install deep_gemm==1.0.0 (-i <IndexURL>)

pip install deep_gemm==1.0.0 (-i <IndexURL>)

deep_gemm

1.0.0

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.8

torch 2.7.0

3.12

oe

pip install deep_gemm==1.0.0 (-i <IndexURL>)

pip install deep_gemm==1.0.0 (-i <IndexURL>)

flash_attn

2.7.4.post1

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.10

oe

pip install flash_attn==2.7.4.post1 (-i <IndexURL>)

pip install flash_attn==2.7.4.post1 (-i <IndexURL>)

flash_attn

2.7.4.post1

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.12

oe

pip install flash_attn==2.7.4.post1 (-i <IndexURL>)

pip install flash_attn==2.7.4.post1 (-i <IndexURL>)

flash_attn

2.7.4.post1

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.8

torch 2.7.0

3.12

oe

pip install flash_attn==2.7.4.post1 (-i <IndexURL>)

pip install flash_attn==2.7.4.post1 (-i <IndexURL>)

flash_attn

2.7.2

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.10

oe

pip install flash_attn==2.7.2 (-i <IndexURL>)

pip install flash_attn==2.7.2 (-i <IndexURL>)

flash_attn

2.7.2

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.12

oe

pip install flash_attn==2.7.2 (-i <IndexURL>)

pip install flash_attn==2.7.2 (-i <IndexURL>)

flash_attn

2.7.2

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.8

torch 2.7.0

3.12

oe

pip install flash_attn==2.7.2 (-i <IndexURL>)

pip install flash_attn==2.7.2 (-i <IndexURL>)

flash_mla

1.0.0

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.10

oe

pip install flash_mla==1.0.0 (-i <IndexURL>)

pip install flash_mla==1.0.0 (-i <IndexURL>)

flash_mla

1.0.0

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.12

oe

pip install flash_mla==1.0.0 (-i <IndexURL>)

pip install flash_mla==1.0.0 (-i <IndexURL>)

flash_mla

1.0.0

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.8

torch 2.7.0

3.12

oe

pip install flash_mla==1.0.0 (-i <IndexURL>)

pip install flash_mla==1.0.0 (-i <IndexURL>)

flashinfer_python

0.2.7.post1

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.10

oe

pip install flashinfer_python==0.2.7.post1 (-i <IndexURL>)

pip install flashinfer_python==0.2.7.post1 (-i <IndexURL>)

flashinfer_python

0.2.7.post1

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.12

oe

pip install flashinfer_python==0.2.7.post1 (-i <IndexURL>)

pip install flashinfer_python==0.2.7.post1 (-i <IndexURL>)

flashinfer_python

0.2.7.post1

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.8

torch 2.7.0

3.12

oe

pip install flashinfer_python==0.2.7.post1 (-i <IndexURL>)

pip install flashinfer_python==0.2.7.post1 (-i <IndexURL>)

flashinfer_python

0.2.6.post1

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.10

oe

pip install flashinfer_python==0.2.6.post1 (-i <IndexURL>)

pip install flashinfer_python==0.2.6.post1 (-i <IndexURL>)

flashinfer_python

0.2.6.post1

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.12

oe

pip install flashinfer_python==0.2.6.post1 (-i <IndexURL>)

pip install flashinfer_python==0.2.6.post1 (-i <IndexURL>)

flashinfer_python

0.2.6.post1

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.8

torch 2.7.0

3.12

oe

pip install flashinfer_python==0.2.6.post1 (-i <IndexURL>)

pip install flashinfer_python==0.2.6.post1 (-i <IndexURL>)

flashinfer_python

0.2.3

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.10

oe

pip install flashinfer_python==0.2.3 (-i <IndexURL>)

pip install flashinfer_python==0.2.3 (-i <IndexURL>)

flashinfer_python

0.2.3

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.12

oe

pip install flashinfer_python==0.2.3 (-i <IndexURL>)

pip install flashinfer_python==0.2.3 (-i <IndexURL>)

flashinfer_python

0.2.3

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.8

torch 2.7.0

3.12

oe

pip install flashinfer_python==0.2.3 (-i <IndexURL>)

pip install flashinfer_python==0.2.3 (-i <IndexURL>)

grouped_gemm

1.1.4

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.10

oe

pip install grouped_gemm==1.1.4 (-i <IndexURL>)

pip install grouped_gemm==1.1.4 (-i <IndexURL>)

grouped_gemm

1.1.4

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.12

oe

pip install grouped_gemm==1.1.4 (-i <IndexURL>)

pip install grouped_gemm==1.1.4 (-i <IndexURL>)

grouped_gemm

1.1.4

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.8

torch 2.7.0

3.12

oe

pip install grouped_gemm==1.1.4 (-i <IndexURL>)

pip install grouped_gemm==1.1.4 (-i <IndexURL>)

jaxlib

0.4.34

AliOS 7.2

CUDA 12.6

3.10

oe

jaxlib

0.4.34

Ubuntu 24.04

CUDA 12.6

3.12

oe

pip install jaxlib==0.4.34 (-i <IndexURL>)

jaxlib

0.4.34

AliOS 7.2

CUDA 12.6

3.12

oe

jaxlib

0.4.34

AliOS 7.2

CUDA 12.8

3.12

oe

jaxlib

0.4.34

Ubuntu 24.04

CUDA 12.8

3.12

oe

pip install jaxlib==0.4.34 (-i <IndexURL>)

jax_cuda12_plugin

0.4.34

AliOS 7.2

CUDA 12.6

3.10

oe

jax_cuda12_plugin

0.4.34

Ubuntu 24.04

CUDA 12.6

3.12

oe

pip install jax_cuda12_plugin==0.4.34 (-i <IndexURL>)

jax_cuda12_plugin

0.4.34

AliOS 7.2

CUDA 12.6

3.12

oe

jax_cuda12_plugin

0.4.34

AliOS 7.2

CUDA 12.8

3.12

oe

jax_cuda12_plugin

0.4.34

Ubuntu 24.04

CUDA 12.8

3.12

oe

pip install jax_cuda12_plugin==0.4.34 (-i <IndexURL>)

jax_cuda12_pjrt

0.4.34

AliOS 7.2

CUDA 12.6

3.10

oe

jax_cuda12_pjrt

0.4.34

Ubuntu 24.04

CUDA 12.6

3.12

oe

pip install jax_cuda12_pjrt==0.4.34 (-i <IndexURL>)

jax_cuda12_pjrt

0.4.34

AliOS 7.2

CUDA 12.6

3.12

oe

jax_cuda12_pjrt

0.4.34

AliOS 7.2

CUDA 12.8

3.12

oe

jax_cuda12_pjrt

0.4.34

Ubuntu 24.04

CUDA 12.8

3.12

oe

pip install jax_cuda12_pjrt==0.4.34 (-i <IndexURL>)

onnxruntime_gpu

1.20.0

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

3.10

oe

pip install onnxruntime_gpu==1.20.0 (-i <IndexURL>)

pip install onnxruntime_gpu==1.20.0 (-i <IndexURL>)

onnxruntime_gpu

1.20.0

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

3.12

oe

pip install onnxruntime_gpu==1.20.0 (-i <IndexURL>)

pip install onnxruntime_gpu==1.20.0 (-i <IndexURL>)

onnxruntime_gpu

1.20.0

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.8

3.12

oe

pip install onnxruntime_gpu==1.20.0 (-i <IndexURL>)

pip install onnxruntime_gpu==1.20.0 (-i <IndexURL>)

onnxruntime_gpu

1.20.1

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

3.10

oe

pip install onnxruntime_gpu==1.20.1 (-i <IndexURL>)

pip install onnxruntime_gpu==1.20.1 (-i <IndexURL>)

onnxruntime_gpu

1.20.1

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

3.12

oe

pip install onnxruntime_gpu==1.20.1 (-i <IndexURL>)

pip install onnxruntime_gpu==1.20.1 (-i <IndexURL>)

onnxruntime_gpu

1.20.1

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.8

3.12

oe

pip install onnxruntime_gpu==1.20.1 (-i <IndexURL>)

pip install onnxruntime_gpu==1.20.1 (-i <IndexURL>)

onnxruntime_training

1.19.0

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

3.10

oe

pip install onnxruntime_training==1.19.0 (-i <IndexURL>)

pip install onnxruntime_training==1.19.0 (-i <IndexURL>)

onnxruntime_training

1.19.0

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

3.12

oe

pip install onnxruntime_training==1.19.0 (-i <IndexURL>)

pip install onnxruntime_training==1.19.0 (-i <IndexURL>)

onnxruntime_training

1.19.2

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

3.10

oe

pip install onnxruntime_training==1.19.2 (-i <IndexURL>)

pip install onnxruntime_training==1.19.2 (-i <IndexURL>)

onnxruntime_training

1.19.2

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

3.12

oe

pip install onnxruntime_training==1.19.2 (-i <IndexURL>)

pip install onnxruntime_training==1.19.2 (-i <IndexURL>)

sglang

0.4.6.post1

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.10

oe

pip install sglang==0.4.6.post1 (-i <IndexURL>)

pip install sglang==0.4.6.post1 (-i <IndexURL>)

sglang

0.4.6.post1

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.12

oe

pip install sglang==0.4.6.post1 (-i <IndexURL>)

pip install sglang==0.4.6.post1 (-i <IndexURL>)

sglang

0.4.6.post1

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.8

torch 2.7.0

3.12

oe

pip install sglang==0.4.6.post1 (-i <IndexURL>)

pip install sglang==0.4.6.post1 (-i <IndexURL>)

sglang

0.4.7

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.10

oe

pip install sglang==0.4.7 (-i <IndexURL>)

pip install sglang==0.4.7 (-i <IndexURL>)

sglang

0.4.7

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.12

oe

pip install sglang==0.4.7 (-i <IndexURL>)

pip install sglang==0.4.7 (-i <IndexURL>)

sglang

0.4.7

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.8

torch 2.7.0

3.12

oe

pip install sglang==0.4.7 (-i <IndexURL>)

pip install sglang==0.4.7 (-i <IndexURL>)

sglang

0.4.9

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.10

oe

pip install sglang==0.4.9 (-i <IndexURL>)

pip install sglang==0.4.9 (-i <IndexURL>)

sglang

0.4.9

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.12

oe

pip install sglang==0.4.9 (-i <IndexURL>)

pip install sglang==0.4.9 (-i <IndexURL>)

sglang

0.4.9

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.8

torch 2.7.0

3.12

oe

pip install sglang==0.4.9 (-i <IndexURL>)

pip install sglang==0.4.9 (-i <IndexURL>)

sgl_kernel

0.1.0

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.10

oe

pip install sgl_kernel==0.1.0 (-i <IndexURL>)

pip install sgl_kernel==0.1.0 (-i <IndexURL>)

sgl_kernel

0.1.0

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.12

oe

pip install sgl_kernel==0.1.0 (-i <IndexURL>)

pip install sgl_kernel==0.1.0 (-i <IndexURL>)

sgl_kernel

0.1.0

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.8

torch 2.7.0

3.12

oe

pip install sgl_kernel==0.1.0 (-i <IndexURL>)

pip install sgl_kernel==0.1.0 (-i <IndexURL>)

sgl_kernel

0.1.7

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.10

oe

pip install sgl_kernel==0.1.7 (-i <IndexURL>)

pip install sgl_kernel==0.1.7 (-i <IndexURL>)

sgl_kernel

0.1.7

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.12

oe

pip install sgl_kernel==0.1.7 (-i <IndexURL>)

pip install sgl_kernel==0.1.7 (-i <IndexURL>)

sgl_kernel

0.1.7

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.8

torch 2.7.0

3.12

oe

pip install sgl_kernel==0.1.7 (-i <IndexURL>)

pip install sgl_kernel==0.1.7 (-i <IndexURL>)

sgl_kernel

0.2.4

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.10

oe

pip install sgl_kernel==0.2.4 (-i <IndexURL>)

pip install sgl_kernel==0.2.4 (-i <IndexURL>)

sgl_kernel

0.2.4

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.12

oe

pip install sgl_kernel==0.2.4 (-i <IndexURL>)

pip install sgl_kernel==0.2.4 (-i <IndexURL>)

sgl_kernel

0.2.4

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.8

torch 2.7.0

3.12

oe

pip install sgl_kernel==0.2.4 (-i <IndexURL>)

pip install sgl_kernel==0.2.4 (-i <IndexURL>)

tensorflow

2.17.0

Ubuntu 22.04

CUDA 12.6

tensorflow 2.17.0

3.10

oe

pip install tensorflow==2.17.0 (-i <IndexURL>)

tensorflow

2.17.0

Ubuntu 24.04

CUDA 12.6

tensorflow 2.17.0

3.12

oe

pip install tensorflow==2.17.0 (-i <IndexURL>)

torch

2.5.1

Ubuntu 22.04, Ubuntu 24.04

cuda12.4

torch 2.5.1

3.10

oe

pip install torch==2.5.1 (-i <IndexURL>)

pip install torch==2.5.1 (-i <IndexURL>)

torch

2.5.1

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.5.1

3.10

oe

pip install torch==2.5.1 (-i <IndexURL>)

pip install torch==2.5.1 (-i <IndexURL>)

torch

2.5.1

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.5.1

3.12

oe

pip install torch==2.5.1 (-i <IndexURL>)

pip install torch==2.5.1 (-i <IndexURL>)

torch

2.6.0

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.10

oe

pip install torch==2.6.0 (-i <IndexURL>)

pip install torch==2.6.0 (-i <IndexURL>)

torch

2.6.0

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.12

oe

pip install torch==2.6.0 (-i <IndexURL>)

pip install torch==2.6.0 (-i <IndexURL>)

torch

2.6.0

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.8

torch 2.6.0

3.12

oe

pip install torch==2.6.0 (-i <IndexURL>)

pip install torch==2.6.0 (-i <IndexURL>)

torch

2.7.0

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.7.0

3.10

oe

pip install torch==2.7.0 (-i <IndexURL>)

pip install torch==2.7.0 (-i <IndexURL>)

torch

2.7.0

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.7.0

3.12

oe

pip install torch==2.7.0 (-i <IndexURL>)

pip install torch==2.7.0 (-i <IndexURL>)

torch

2.7.0

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.8

torch 2.7.0

3.12

oe

pip install torch==2.7.0 (-i <IndexURL>)

pip install torch==2.7.0 (-i <IndexURL>)

torchao

0.11.0

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.10

oe

pip install torchao==0.11.0 (-i <IndexURL>)

pip install torchao==0.11.0 (-i <IndexURL>)

torchao

0.11.0

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.12

oe

pip install torchao==0.11.0 (-i <IndexURL>)

pip install torchao==0.11.0 (-i <IndexURL>)

torchao

0.11.0

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.8

torch 2.7.0

3.12

oe

pip install torchao==0.11.0 (-i <IndexURL>)

pip install torchao==0.11.0 (-i <IndexURL>)

transformer_engine

2.2

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.10

oe

pip install transformer_engine==2.2 (-i <IndexURL>)

pip install transformer_engine==2.2 (-i <IndexURL>)

transformer_engine

2.2

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.12

oe

pip install transformer_engine==2.2 (-i <IndexURL>)

pip install transformer_engine==2.2 (-i <IndexURL>)

transformer_engine

2.2

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.8

torch 2.7.0

3.12

oe

pip install transformer_engine==2.2 (-i <IndexURL>)

pip install transformer_engine==2.2 (-i <IndexURL>)

transformer_engine

2.3

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.10

oe

pip install transformer_engine==2.3 (-i <IndexURL>)

pip install transformer_engine==2.3 (-i <IndexURL>)

transformer_engine

2.3

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.12

oe

pip install transformer_engine==2.3 (-i <IndexURL>)

pip install transformer_engine==2.3 (-i <IndexURL>)

transformer_engine

2.3

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.8

torch 2.7.0

3.12

oe

pip install transformer_engine==2.3 (-i <IndexURL>)

pip install transformer_engine==2.3 (-i <IndexURL>)

triton

3.0.0

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

3.10

oe

pip install triton==3.0.0 (-i <IndexURL>)

pip install triton==3.0.0 (-i <IndexURL>)

triton

3.0.0

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

3.12

oe

pip install triton==3.0.0 (-i <IndexURL>)

pip install triton==3.0.0 (-i <IndexURL>)

triton

3.0.0

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.8

3.12

oe

pip install triton==3.0.0 (-i <IndexURL>)

pip install triton==3.0.0 (-i <IndexURL>)

triton

3.1.0

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

3.10

oe

pip install triton==3.1.0 (-i <IndexURL>)

pip install triton==3.1.0 (-i <IndexURL>)

triton

3.1.0

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

3.12

oe

pip install triton==3.1.0 (-i <IndexURL>)

pip install triton==3.1.0 (-i <IndexURL>)

triton

3.1.0

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.8

3.12

oe

pip install triton==3.1.0 (-i <IndexURL>)

pip install triton==3.1.0 (-i <IndexURL>)

triton

3.2.0

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

3.10

oe

pip install triton==3.2.0 (-i <IndexURL>)

pip install triton==3.2.0 (-i <IndexURL>)

triton

3.2.0

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

3.12

oe

pip install triton==3.2.0 (-i <IndexURL>)

pip install triton==3.2.0 (-i <IndexURL>)

triton

3.2.0

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.8

3.12

oe

pip install triton==3.2.0 (-i <IndexURL>)

pip install triton==3.2.0 (-i <IndexURL>)

triton

3.3.0

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

3.10

oe

pip install triton==3.3.0 (-i <IndexURL>)

pip install triton==3.3.0 (-i <IndexURL>)

triton

3.3.0

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

3.12

oe

pip install triton==3.3.0 (-i <IndexURL>)

pip install triton==3.3.0 (-i <IndexURL>)

triton

3.3.0

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.8

3.12

oe

pip install triton==3.3.0 (-i <IndexURL>)

pip install triton==3.3.0 (-i <IndexURL>)

vllm_flash_attn

2.6.2

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.10

oe

pip install vllm_flash_attn==2.6.2 (-i <IndexURL>)

pip install vllm_flash_attn==2.6.2 (-i <IndexURL>)

vllm_flash_attn

2.6.2

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.12

oe

pip install vllm_flash_attn==2.6.2 (-i <IndexURL>)

pip install vllm_flash_attn==2.6.2 (-i <IndexURL>)

vllm_flash_attn

2.6.2

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.8

torch 2.7.0

3.12

oe

pip install vllm_flash_attn==2.6.2 (-i <IndexURL>)

pip install vllm_flash_attn==2.6.2 (-i <IndexURL>)

vllm

0.9.1

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.10

oe

pip install vllm==0.9.1 (-i <IndexURL>)

pip install vllm==0.9.1 (-i <IndexURL>)

vllm

0.9.1

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.12

oe

pip install vllm==0.9.1 (-i <IndexURL>)

pip install vllm==0.9.1 (-i <IndexURL>)

vllm

0.9.1

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.8

torch 2.7.0

3.12

oe

pip install vllm==0.9.1 (-i <IndexURL>)

pip install vllm==0.9.1 (-i <IndexURL>)

xgboost

3.0.2

AliOS 7.2

CUDA 12.6

3.10

ce

xgboost

3.0.2

Ubuntu 22.04

CUDA 12.6

3.10

ce

pip install xgboost==3.0.2 (-i <IndexURL>)

xgboost

3.0.2

AliOS 7.2

CUDA 12.6

3.12

ce

xgboost

3.0.2

Ubuntu 24.04

CUDA 12.6

3.12

ce

pip install xgboost==3.0.2 (-i <IndexURL>)

xgboost

3.0.2

AliOS 7.2

CUDA 12.8

3.12

ce

xgboost

3.0.2

Ubuntu 24.04

CUDA 12.8

3.12

ce

pip install xgboost==3.0.2 (-i <IndexURL>)

xformers

0.0.30

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.10

oe

pip install xformers==0.0.30 (-i <IndexURL>)

pip install xformers==0.0.30 (-i <IndexURL>)

xformers

0.0.30

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.12

oe

pip install xformers==0.0.30 (-i <IndexURL>)

pip install xformers==0.0.30 (-i <IndexURL>)

xformers

0.0.30

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.8

torch 2.7.0

3.12

oe

pip install xformers==0.0.30 (-i <IndexURL>)

pip install xformers==0.0.30 (-i <IndexURL>)

xformers

0.0.29.post1

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.10

oe

pip install xformers==0.0.29.post1 (-i <IndexURL>)

pip install xformers==0.0.29.post1 (-i <IndexURL>)

xformers

0.0.29.post1

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.6

torch 2.6.0

3.12

oe

pip install xformers==0.0.29.post1 (-i <IndexURL>)

pip install xformers==0.0.29.post1 (-i <IndexURL>)

xformers

0.0.29.post1

Ubuntu 22.04, Ubuntu 24.04

CUDA 12.8

torch 2.7.0

3.12

oe

pip install xformers==0.0.29.post1 (-i <IndexURL>)

pip install xformers==0.0.29.post1 (-i <IndexURL>)