From f989495458d0655d8acfd93d5a361c23469c5ad0 Mon Sep 17 00:00:00 2001
From: Tongze Wang <wtongze@gmail.com>
Date: Thu, 12 Sep 2024 13:20:48 -0700
Subject: [PATCH] fix: vision dep

---
 pytorch.dockerfile | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/pytorch.dockerfile b/pytorch.dockerfile
index 875f91c..295ab5d 100644
--- a/pytorch.dockerfile
+++ b/pytorch.dockerfile
@@ -50,7 +50,7 @@ PYTORCH_BUILD_VERSION=$(echo "${PYTORCH_BRANCH}" | tr -d "v") python${PYTHON_VER
 
 RUN <<EOF
 sudo apt-get update
-sudo apt-get install -y ffmpeg
+sudo apt-get install -y ffmpeg libjpeg-dev libpng-dev
 sudo apt-get clean
 sudo rm -rf /var/lib/apt/lists/*
 EOF