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) # 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 # Build parallemism
ARG BUILD_JOBS ARG BUILD_JOBS