diff --git a/Dockerfile b/Dockerfile index 210cb6e033fc137a5e905f75fe97815568e8885a..fed359e944e9638aadf62c2cfd0027dac4c113ef 100644 --- a/Dockerfile +++ b/Dockerfile @@ -1 +1 @@ -FROM nvidia/12.3.0-base-ubuntu20.04 +FROM nvidia/cuda:12.3.0-base-ubuntu20.04