From 83f2afdb204fa5697ee22af5f4d72b872a9c87a1 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Mikel=20Brostr=C3=B6m?= <mikel.brostrom@cevt.se>
Date: Tue, 9 Jul 2024 14:38:28 +0200
Subject: [PATCH] test lighter image

---
 Dockerfile | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/Dockerfile b/Dockerfile
index 61abe648d4..d693c69301 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -2,7 +2,7 @@
 #   docker build -t mikel-brostrom/yolov5_strongsort_osnet .
 
 # Base image: Nvidia PyTorch https://ngc.nvidia.com/catalog/containers/nvidia:pytorch
-FROM nvcr.io/nvidia/pytorch:24.06-py3
+FROM pytorch/pytorch:2.3.1-cuda11.8-cudnn8-runtime
 
 # Update image
 RUN apt update