Rolling back base image

This commit is contained in:
Eugene Rakhmatulin
2026-03-04 07:19:43 -08:00
parent ae19b66fdd
commit fff1a24982

View File

@@ -6,7 +6,7 @@ ARG BUILD_JOBS=16
# =========================================================
# STAGE 1: Base Image (Installs Dependencies)
# =========================================================
FROM nvcr.io/nvidia/pytorch:26.02-py3 AS base
FROM nvcr.io/nvidia/pytorch:26.01-py3 AS base
# Build parallemism
ARG BUILD_JOBS