feat(containerization): support 11.7 (#3567)

Signed-off-by: Aaron Pham <29749331+aarnphm@users.noreply.github.com>
bentoml · Feb 15, 2023 · dce24d8 · dce24d8
1 parent 97b903d
commit dce24d8
Showing 1 changed file with 4 additions and 2 deletions.
diff --git 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",