diff --git a/src/bentoml/_internal/container/frontend/dockerfile/__init__.py b/src/bentoml/_internal/container/frontend/dockerfile/__init__.py index 4718adbd3a7..a91136ffc21 100644 --- a/src/bentoml/_internal/container/frontend/dockerfile/__init__.py +++ b/src/bentoml/_internal/container/frontend/dockerfile/__init__.py @@ -23,10 +23,12 @@ # Python supported versions SUPPORTED_PYTHON_VERSIONS = ["3.7", "3.8", "3.9", "3.10"] # CUDA supported versions -SUPPORTED_CUDA_VERSIONS = ["11.6.2", "11.4.3", "11.2.2"] +SUPPORTED_CUDA_VERSIONS = ["11.7.0", "11.6.2", "11.4.3", "11.2.2"] # Mapping from user provided version argument to the full version target to install ALLOWED_CUDA_VERSION_ARGS = { - "11": "11.6.2", + "11": "11.7.0", + "11.7": "11.7.0", + "11.7.0": "11.7.0", "11.6": "11.6.2", "11.6.2": "11.6.2", "11.4": "11.4.3",