diff --git a/Dockerfile b/Dockerfile
index ab2e2f5..5ef595b 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,9 +1,9 @@
-ARG CUDA_VER=11.4.0
-ARG LINUX_VER=ubuntu20.04
+ARG CUDA_VER=11.8.0
+ARG LINUX_VER=ubuntu22.04
 FROM nvidia/cuda:${CUDA_VER}-base-${LINUX_VER}
 
 ARG LINUX_VER
-ARG PYTHON_VER=3.9
+ARG PYTHON_VER=3.10
 ARG DEBIAN_FRONTEND=noninteractive
 ENV PATH=/opt/conda/bin:$PATH
 ENV PYTHON_VERSION=${PYTHON_VER}
diff --git a/axis.yaml b/axis.yaml
index 10c84fb..19f3281 100644
--- a/axis.yaml
+++ b/axis.yaml
@@ -4,8 +4,8 @@ CUDA_VER:
   - "11.5.1"
   - "11.8.0"
   - "12.0.1"
+  - "12.1.1"
 PYTHON_VER:
-  - "3.8"
   - "3.9"
   - "3.10"
 LINUX_VER: