From 3731ee08839f449af0225d456f57e0af6a2edfb4 Mon Sep 17 00:00:00 2001 From: Dipika Sikka Date: Tue, 5 Mar 2024 16:48:06 +0000 Subject: [PATCH] update --- docker/containers/docker_nightly/Dockerfile | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/docker/containers/docker_nightly/Dockerfile b/docker/containers/docker_nightly/Dockerfile index 6e58823818b..22a323018cc 100644 --- a/docker/containers/docker_nightly/Dockerfile +++ b/docker/containers/docker_nightly/Dockerfile @@ -1,5 +1,6 @@ ARG SOURCE=ghcr.io/neuralmagic/cuda-python3.8 -ARG TORCH_VERSION=1.10.1 +# how were these versions selected before? +ARG TORCH_VERSION=1.10.2 ARG TORCHVISION_VERSION=0.11.3 ARG CUDA=113