From 19beb247f08f437124fd8b6b1520190becd19884 Mon Sep 17 00:00:00 2001 From: hailin Date: Tue, 1 Jul 2025 11:51:14 +0800 Subject: [PATCH] . --- Dockerfile | 19 +++++++++++-------- 1 file changed, 11 insertions(+), 8 deletions(-) diff --git a/Dockerfile b/Dockerfile index 6ac4319..15c7889 100644 --- a/Dockerfile +++ b/Dockerfile @@ -10,11 +10,12 @@ ENV DEBIAN_FRONTEND=noninteractive \ USE_CUDA=1 USE_DISTRIBUTED=0 BUILD_TEST=0 TORCH_CUDA_ARCH_LIST="7.5;8.0;8.6;9.0" RUN apt-get update && apt-get install -y --no-install-recommends \ - python3 python3-pip python3-distutils \ - git cmake ninja-build \ - libopenblas-dev libopenmpi-dev libnccl-dev \ - libjpeg-dev libpng-dev ca-certificates && \ - python3 -m pip install --no-cache-dir --upgrade pip wheel setuptools sympy + python3 python3-pip python3-distutils git cmake ninja-build \ + libopenblas-dev libopenmpi-dev \ + libnccl2=2.22.3-1+cuda12.6 \ + libnccl-dev=2.22.3-1+cuda12.6 \ + libjpeg-dev libpng-dev ca-certificates && \ + python3 -m pip install --no-cache-dir --upgrade pip wheel setuptoo WORKDIR /opt RUN git clone --recursive -b v2.7.1 https://github.com/pytorch/pytorch.git @@ -32,9 +33,11 @@ FROM nvidia/cuda:${CUDA_VERSION}-devel-ubuntu22.04 AS builder-extras ENV DEBIAN_FRONTEND=noninteractive PYTHONUNBUFFERED=1 LANG=C.UTF-8 LC_ALL=C.UTF-8 RUN apt-get update && apt-get install -y --no-install-recommends \ - python3 python3-pip python3-distutils \ - git build-essential cmake ninja-build \ - libjpeg-dev libpng-dev ca-certificates && \ + python3 python3-pip python3-distutils git build-essential \ + cmake ninja-build libjpeg-dev libpng-dev ca-certificates \ + libopenmpi-dev \ + libnccl2=2.22.3-1+cuda12.6 \ + libnccl-dev=2.22.3-1+cuda12.6 && \ python3 -m pip install --no-cache-dir --upgrade pip wheel setuptools # ── 安装自编 torch 轮子 ──────────────────────────────────────────────────────