From dce24d8b868143a0279fbf886fb1c71ccb80a87d Mon Sep 17 00:00:00 2001
From: Aaron Pham <29749331+aarnphm@users.noreply.github.com>
Date: Tue, 14 Feb 2023 20:47:32 -0800
Subject: [PATCH] feat(containerization): support 11.7 (#3567)

Signed-off-by: Aaron Pham <29749331+aarnphm@users.noreply.github.com>
---
 .../_internal/container/frontend/dockerfile/__init__.py     | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

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",