From 32087331aeb6523b3fd53d9179f6152020ebc341 Mon Sep 17 00:00:00 2001 From: Rayed Bin Wahed Date: Fri, 1 Mar 2024 02:10:01 +0800 Subject: [PATCH] Update Dockerfile Use significantly smaller docker image --- .devcontainer/Dockerfile | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.devcontainer/Dockerfile b/.devcontainer/Dockerfile index cc4327b..79e7f40 100644 --- a/.devcontainer/Dockerfile +++ b/.devcontainer/Dockerfile @@ -1,4 +1,4 @@ -FROM pytorch/pytorch:2.2.0-cuda12.1-cudnn8-devel +FROM pytorch/pytorch:2.2.0-cuda12.1-cudnn8-runtime RUN <