diff --git a/src/sample_pytorch_gpu_project/.devcontainer/Dockerfile b/src/sample_pytorch_gpu_project/.devcontainer/Dockerfile index 95013c2..fa7a7c9 100644 --- a/src/sample_pytorch_gpu_project/.devcontainer/Dockerfile +++ b/src/sample_pytorch_gpu_project/.devcontainer/Dockerfile @@ -1,5 +1,5 @@ # leverage multi-stage build and copy python binaries to cuda container -FROM python:3.11.10 as python +FROM python:3.11.11 as python FROM nvidia/cuda:12.6.2-cudnn-runtime-ubuntu24.04 COPY --from=python /usr/local/ /usr/local/