From b87efae88c02510178f5640430d467969816d869 Mon Sep 17 00:00:00 2001 From: Binyang Li Date: Sat, 7 Dec 2024 08:03:05 +0000 Subject: [PATCH] Fix --- docker/base-dev-x.dockerfile | 1 + python/requirements_cuda12.txt | 2 -- 2 files changed, 1 insertion(+), 2 deletions(-) diff --git a/docker/base-dev-x.dockerfile b/docker/base-dev-x.dockerfile index 5aeaa4142..26216711e 100644 --- a/docker/base-dev-x.dockerfile +++ b/docker/base-dev-x.dockerfile @@ -28,6 +28,7 @@ ADD . /tmp/mscclpp WORKDIR /tmp/mscclpp ARG TARGET="cuda12.1" RUN target_type=$(echo $TARGET | sed 's/\.[0-9]*$//') && \ + python3 -m pip install --no-cache-dir --upgrade pip && \ python3 -m pip install --no-cache-dir -r python/requirements_${target_type}.txt # Set PATH diff --git a/python/requirements_cuda12.txt b/python/requirements_cuda12.txt index 445482bd6..71014a544 100644 --- a/python/requirements_cuda12.txt +++ b/python/requirements_cuda12.txt @@ -5,5 +5,3 @@ netifaces pytest numpy matplotlib -# scikit-build issue, need to fix -ninja