From 209ab7d013b278fda8e3a4a8a3a113fdec31749a Mon Sep 17 00:00:00 2001 From: Nicolas Patry Date: Thu, 3 Aug 2023 14:19:12 +0000 Subject: [PATCH] This should prevent the PyTorch overriding. --- Dockerfile | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/Dockerfile b/Dockerfile index 679fb48a..9d3d6fe0 100644 --- a/Dockerfile +++ b/Dockerfile @@ -39,7 +39,7 @@ RUN cargo build --release # Adapted from: https://github.com/pytorch/pytorch/blob/master/Dockerfile FROM debian:bullseye-slim as pytorch-install -ARG PYTORCH_VERSION=2.0.0 +ARG PYTORCH_VERSION=2.0.1 ARG PYTHON_VERSION=3.9 ARG CUDA_VERSION=11.8 ARG MAMBA_VERSION=23.1.0-1