diff --git a/Dockerfile b/Dockerfile index fa2a257..6a5ad92 100644 --- a/Dockerfile +++ b/Dockerfile @@ -1,5 +1,5 @@ ARG CUDA_VERSION=12.8.1 -ARG PYTHON_VERSION=3.12 +ARG PYTHON_VERSION=3.10 ARG BUILD_BASE_IMAGE=nvidia/cuda:${CUDA_VERSION}-devel-ubuntu20.04