diff --git a/Dockerfile b/Dockerfile index 7c819a1..00fd302 100644 --- a/Dockerfile +++ b/Dockerfile @@ -1,4 +1,4 @@ -FROM nvidia/cuda:12.2.2-cudnn8-runtime-ubuntu22.04 +FROM nvidia/cuda:11.7.1-cudnn8-runtime-ubuntu20.04 WORKDIR /subgen