From a65f6b9e9b6375395fb60eabbe0a29cd2007fc62 Mon Sep 17 00:00:00 2001
From: Logan Adams <114770087+loadams@users.noreply.github.com>
Date: Fri, 9 Jun 2023 12:31:03 -0700
Subject: [PATCH] Update Dockerfile with newer cuda and torch. (#3716)

* Add non-interactive prompt, causing issues for some users

* Update pytorch version too
---
 docker/Dockerfile | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/docker/Dockerfile b/docker/Dockerfile
index cc1d792fb..d8bd95427 100755
--- a/docker/Dockerfile
+++ b/docker/Dockerfile
@@ -1,4 +1,6 @@
-FROM nvidia/cuda:10.0-devel-ubuntu18.04
+FROM nvidia/cuda:11.7.0-devel-ubuntu18.04
+
+ENV DEBIAN_FRONTEND noninteractive
 
 ##############################################################################
 # Temporary Installation Directory
@@ -157,8 +159,8 @@ RUN cat /etc/ssh/sshd_config > ${STAGE_DIR}/sshd_config && \
 ##############################################################################
 # PyTorch
 ##############################################################################
-ENV PYTORCH_VERSION=1.2.0
-ENV TORCHVISION_VERSION=0.4.0
+ENV PYTORCH_VERSION=1.9.0
+ENV TORCHVISION_VERSION=0.10.0
 ENV TENSORBOARDX_VERSION=1.8
 RUN pip install torch==${PYTORCH_VERSION}
 RUN pip install torchvision==${TORCHVISION_VERSION}