diff --git a/docker/sgl-deep-gemm.Dockerfile b/docker/sgl-deep-gemm.Dockerfile index 0083d4ddc..5ad22c60e 100644 --- a/docker/sgl-deep-gemm.Dockerfile +++ b/docker/sgl-deep-gemm.Dockerfile @@ -8,7 +8,7 @@ ARG CUDA_VERSION=13.0 ARG PYTHON_VERSION=3.12 ARG PYTHON_TAG=cp312-cp312 ARG TORCH_VER=2.11.0 -ARG TVM_FFI_VER=0.1.9 +ARG TVM_FFI_VER=0.1.11 ARG PIP_DEFAULT_INDEX=https://pypi.python.org/simple ARG PYTORCH_MIRROR=download.pytorch.org