Skip to content

Commit db6ce1c

Browse files
authored
Merge pull request pytorch#168 from pytorch/adamlouly/fix_nightly_pipeline
upgrade torch vision
2 parents 61c2676 + 75cf8b9 commit db6ce1c

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

torch_ort/docker/Dockerfile.ort-default-stable-torch1.13.1-onnxruntime190-cu116-cudnn8-devel-ubuntu18.04

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@ FROM nvidia/cuda:11.6.2-cudnn8-devel-ubuntu18.04 as builder
77
ARG TORCH_CUDA_VERSION=cu116
88
ARG ONNXRUNTIME_TRAINING_VERSION=1.12.0
99
ARG TORCH_VERSION=1.13.1
10-
ARG TORCHVISION_VERSION=0.13.1
10+
ARG TORCHVISION_VERSION=0.14.1
1111

1212
# Install and update tools to minimize security vulnerabilities
1313
RUN apt-get update

torch_ort/docker/Dockerfile.ort-torch1.13.1-onnxruntime-nightly-cu116-cudnn8-devel-ubuntu18.04

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@ FROM nvidia/cuda:11.6.2-cudnn8-devel-ubuntu18.04 as builder
77

88
ARG TORCH_CUDA_VERSION=cu116
99
ARG TORCH_VERSION=1.13.1
10-
ARG TORCHVISION_VERSION=0.13.1
10+
ARG TORCHVISION_VERSION=0.14.1
1111

1212
# Install and update tools to minimize security vulnerabilities
1313
RUN apt-get update

0 commit comments

Comments
 (0)