diff --git a/.github/workflows/docgen.yml b/.github/workflows/docgen.yml index fc6afa8d0d..a943efe302 100644 --- a/.github/workflows/docgen.yml +++ b/.github/workflows/docgen.yml @@ -14,12 +14,12 @@ jobs: if: ${{ ! contains(github.actor, 'pytorchbot') }} environment: pytorchbot-env container: - image: docker.io/pytorch/manylinux2_28-builder:cuda12.8 + image: docker.io/pytorch/manylinux2_28-builder:cuda12.9 options: --gpus all env: - CUDA_HOME: /usr/local/cuda-12.8 - VERSION_SUFFIX: cu128 - CU_VERSION: cu128 + CUDA_HOME: /usr/local/cuda-12.9 + VERSION_SUFFIX: cu129 + CU_VERSION: cu129 CHANNEL: nightly CI_BUILD: 1 steps: @@ -35,14 +35,14 @@ jobs: - name: Install base deps run: | python3 -m pip install pip --upgrade - python3 -m pip install pyyaml numpy torch --pre --extra-index-url https://download.pytorch.org/whl/nightly/cu128 + python3 -m pip install pyyaml numpy torch --pre --extra-index-url https://download.pytorch.org/whl/nightly/cu129 ./packaging/pre_build_script.sh - name: Get HEAD SHA id: vars run: echo "sha=$(git rev-parse --short HEAD)" >> $GITHUB_OUTPUT - name: Build Python Package run: | - python3 -m pip install --pre . --extra-index-url https://download.pytorch.org/whl/nightly/cu128 + python3 -m pip install --pre . --extra-index-url https://download.pytorch.org/whl/nightly/cu129 - name: Generate New Docs run: | cd docsrc diff --git a/MODULE.bazel b/MODULE.bazel index 59196b85be..6087f5dffb 100644 --- a/MODULE.bazel +++ b/MODULE.bazel @@ -1,6 +1,6 @@ module( name = "torch_tensorrt", - version = "2.8.0a0", + version = "2.9.0a0", repo_name = "org_pytorch_tensorrt", ) diff --git a/cpp/include/torch_tensorrt/macros.h b/cpp/include/torch_tensorrt/macros.h index bdc25f6cd8..020b94c114 100644 --- a/cpp/include/torch_tensorrt/macros.h +++ b/cpp/include/torch_tensorrt/macros.h @@ -24,7 +24,7 @@ #define STR(x) XSTR(x) #define TORCH_TENSORRT_MAJOR_VERSION 2 -#define TORCH_TENSORRT_MINOR_VERSION 6 +#define TORCH_TENSORRT_MINOR_VERSION 9 #define TORCH_TENSORRT_PATCH_VERSION 0 #define TORCH_TENSORRT_VERSION \ STR(TORCH_TENSORRT_MAJOR_VERSION) \ diff --git a/docker/Dockerfile b/docker/Dockerfile index 23786435c6..b218211e38 100644 --- a/docker/Dockerfile +++ b/docker/Dockerfile @@ -2,9 +2,9 @@ # Base image starts with CUDA #TODO: cuda version -ARG BASE_IMG=nvidia/cuda:12.8.0-devel-ubuntu22.04 +ARG BASE_IMG=nvidia/cuda:12.9.0-devel-ubuntu22.04 FROM ${BASE_IMG} as base -ENV BASE_IMG=nvidia/cuda:12.8.0-devel-ubuntu22.04 +ENV BASE_IMG=nvidia/cuda:12.9.0-devel-ubuntu22.04 ARG TENSORRT_VERSION ENV TENSORRT_VERSION=${TENSORRT_VERSION} diff --git a/docker/dist-build.sh b/docker/dist-build.sh index faabedade2..7790c05f82 100755 --- a/docker/dist-build.sh +++ b/docker/dist-build.sh @@ -4,7 +4,7 @@ set -x TOP_DIR=$(cd $(dirname $0); pwd)/.. -BUILD_CMD="python -m pip wheel . --extra-index-url https://download.pytorch.org/whl/nightly/cu128 -w dist" +BUILD_CMD="python -m pip wheel . --extra-index-url https://download.pytorch.org/whl/nightly/cu129 -w dist" # TensorRT restricts our pip version cd ${TOP_DIR} \ diff --git a/py/requirements.txt b/py/requirements.txt index 302b7e92af..a34a458938 100644 --- a/py/requirements.txt +++ b/py/requirements.txt @@ -2,7 +2,7 @@ numpy packaging pybind11==2.6.2 --extra-index-url https://download.pytorch.org/whl/nightly/cu129 -torch>=2.8.0.dev,<2.9.0 +torch>=2.9.0.dev,<2.10.0 --extra-index-url https://pypi.ngc.nvidia.com pyyaml dllist \ No newline at end of file diff --git a/pyproject.toml b/pyproject.toml index b45cd96d5d..f32e3296d0 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -10,7 +10,7 @@ requires = [ "future>=0.18.3", "tensorrt-cu12>=10.11.0,<10.12.0; platform_machine != 'aarch64' or (platform_machine == 'aarch64' and 'tegra' not in platform_release)", "tensorrt-cu12>=10.3.0,<10.4.0; platform_machine == 'aarch64' and 'tegra' in platform_release", - "torch>=2.8.0.dev,<2.9.0; platform_machine != 'aarch64' or (platform_machine == 'aarch64' and 'tegra' not in platform_release)", + "torch>=2.9.0.dev,<2.10.0; platform_machine != 'aarch64' or (platform_machine == 'aarch64' and 'tegra' not in platform_release)", "torch>=2.7.0,<2.8.0; platform_machine == 'aarch64' and 'tegra' in platform_release", "pybind11==2.6.2", "numpy; platform_machine != 'aarch64' or (platform_machine == 'aarch64' and 'tegra' not in platform_release)", @@ -59,7 +59,7 @@ keywords = [ "inference", ] dependencies = [ - "torch>=2.8.0.dev,<2.9.0; platform_machine != 'aarch64' or (platform_machine == 'aarch64' and 'tegra' not in platform_release)", + "torch>=2.9.0.dev,<2.10.0; platform_machine != 'aarch64' or (platform_machine == 'aarch64' and 'tegra' not in platform_release)", "torch>=2.7.0,<2.8.0; platform_machine == 'aarch64' and 'tegra' in platform_release", "tensorrt>=10.11.0,<10.12.0; platform_machine != 'aarch64' or (platform_machine == 'aarch64' and 'tegra' not in platform_release)", @@ -135,17 +135,17 @@ index-strategy = "unsafe-best-match" [tool.uv.sources] torch = [ - { index = "pytorch-nightly-cu128", marker = "platform_machine != 'aarch64' or (platform_machine == 'aarch64' and 'tegra' not in platform_release)" }, + { index = "pytorch-nightly-cu129", marker = "platform_machine != 'aarch64' or (platform_machine == 'aarch64' and 'tegra' not in platform_release)" }, { index = "jetson-containers", marker = "platform_machine == 'aarch64' and 'tegra' in platform_release" }, ] torchvision = [ - { index = "pytorch-nightly-cu128", marker = "platform_machine != 'aarch64' or (platform_machine == 'aarch64' and 'tegra' not in platform_release)" }, + { index = "pytorch-nightly-cu129", marker = "platform_machine != 'aarch64' or (platform_machine == 'aarch64' and 'tegra' not in platform_release)" }, { index = "jetson-containers", marker = "platform_machine == 'aarch64' and 'tegra' in platform_release" }, ] [[tool.uv.index]] -name = "pytorch-nightly-cu128" -url = "https://download.pytorch.org/whl/nightly/cu128" +name = "pytorch-nightly-cu129" +url = "https://download.pytorch.org/whl/nightly/cu129" explicit = false [[tool.uv.index]] diff --git a/version.txt b/version.txt index 11922a5ce1..03e905f0db 100644 --- a/version.txt +++ b/version.txt @@ -1 +1 @@ -2.8.0a0 +2.9.0a0