From 1f20b261f5e5d7892f92a08db8a99462f7298ba9 Mon Sep 17 00:00:00 2001 From: "Peter St. John" Date: Wed, 18 Dec 2024 09:53:28 -0800 Subject: [PATCH] Attempts to bump the base image to pytorch:24.07 --- Dockerfile | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/Dockerfile b/Dockerfile index f013c8fa6..e68e794fc 100644 --- a/Dockerfile +++ b/Dockerfile @@ -1,5 +1,5 @@ # Base image with apex and transformer engine, but without NeMo or Megatron-LM. -ARG BASE_IMAGE=nvcr.io/nvidia/pytorch:24.02-py3 +ARG BASE_IMAGE=nvcr.io/nvidia/pytorch:24.07-py3 FROM rust:1.82.0 as rust-env @@ -91,7 +91,6 @@ ENV UV_LINK_MODE=copy \ # Install the bionemo-geomtric requirements ahead of copying over the rest of the repo, so that we can cache their # installation. These involve building some torch extensions, so they can take a while to install. RUN --mount=type=bind,source=./sub-packages/bionemo-geometric/requirements.txt,target=/requirements-pyg.txt \ - --mount=type=cache,id=uv-cache,target=/root/.cache,sharing=locked \ uv pip install --no-build-isolation -r /requirements-pyg.txt WORKDIR /workspace/bionemo2