diff --git a/CMakeLists.txt b/CMakeLists.txt index 15ff842..3f68c5d 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -32,9 +32,9 @@ set(RAPIDS_TRITON_REPO_PATH "https://github.com/rapidsai/rapids-triton.git" CACH set(RAPIDS_TRITON_REPO_TAG "branch-${RAPIDS_DEPENDENCIES_VERSION}" CACHE STRING "Tag for rapidsai/rapids-triton repo") if(TRITON_FIL_DOCKER_BUILD) - project(RAPIDS_TRITON_BACKEND VERSION 22.10.00) - set(TRITON_BUILD_CONTAINER "nvcr.io/nvidia/tritonserver:23.09-py3" CACHE STRING "Build image for Dockerized builds") - set(TRITON_BUILD_CONTAINER_VERSION "23.09" CACHE STRING "Triton version for Dockerized builds") + project(RAPIDS_TRITON_BACKEND VERSION 24.11.00) + set(TRITON_BUILD_CONTAINER "nvcr.io/nvidia/tritonserver:24.10-py3" CACHE STRING "Build image for Dockerized builds") + set(TRITON_BUILD_CONTAINER_VERSION "24.10" CACHE STRING "Triton version for Dockerized builds") add_custom_command( OUTPUT fil/libtriton_fil.so $<$:fil/libcuml++.so> diff --git a/conda/environments/buildpy.yml b/conda/environments/buildpy.yml index f7eb252..21041c0 100644 --- a/conda/environments/buildpy.yml +++ b/conda/environments/buildpy.yml @@ -5,3 +5,4 @@ channels: dependencies: - docker-py - python + - distro diff --git a/conda/environments/rapids_triton_dev.yml b/conda/environments/rapids_triton_dev.yml index ce367a5..efab12a 100644 --- a/conda/environments/rapids_triton_dev.yml +++ b/conda/environments/rapids_triton_dev.yml @@ -6,6 +6,7 @@ dependencies: - ccache - cmake>=3.26.4,!=3.30.0 - ninja + - python # TODO(hcho3): Remove the pin when # https://github.com/triton-inference-server/common/pull/114 is merged - rapidjson>=1.1.0,<1.1.0.post*