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