From 0417cca1563047d6e108335e923b85d7c3a1f97d Mon Sep 17 00:00:00 2001 From: root Date: Mon, 8 Jul 2024 22:46:27 -0700 Subject: [PATCH 1/2] cudatoolkit version added in meta.yaml build host and run section --- recipe/meta.yaml | 3 +++ 1 file changed, 3 insertions(+) diff --git a/recipe/meta.yaml b/recipe/meta.yaml index 5ea1890..ae1e78b 100644 --- a/recipe/meta.yaml +++ b/recipe/meta.yaml @@ -21,10 +21,12 @@ requirements: - {{ compiler('c') }} - python {{ python }} - pytorch {{ pytorch }} + - cudatoolkit {{ cudatoolkit }} #[build_type == 'cuda'] host: - pip - pytorch {{ pytorch }} - python {{ python }} + - cudatoolkit {{ cudatoolkit }} #[build_type == 'cuda'] run: - hjson-py - ninja {{ ninja }} @@ -36,6 +38,7 @@ requirements: - pydantic {{ pydantic }} - tqdm {{ tqdm }} - pytorch {{ pytorch }} + - cudatoolkit {{ cudatoolkit }} #[build_type == 'cuda'] about: home: https://github.com/microsoft/DeepSpeed.git From da65ed91c091f6df7b4624953a5d958b6c142ef6 Mon Sep 17 00:00:00 2001 From: root Date: Mon, 8 Jul 2024 23:21:58 -0700 Subject: [PATCH 2/2] Incremented build number --- recipe/meta.yaml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/recipe/meta.yaml b/recipe/meta.yaml index ae1e78b..d6d8213 100644 --- a/recipe/meta.yaml +++ b/recipe/meta.yaml @@ -10,7 +10,7 @@ source: git_rev: v{{ version }} build: - number: 1 + number: 2 string: cuda{{ cudatoolkit | replace(".*", "") }}_py{{ python | replace(".", "") }}_{{ PKG_BUILDNUM }} script: DS_BUILD_SPARSE_ATTN=0; DS_BUILD_OPS=1; {{ PYTHON }} -m pip install . --no-deps --ignore-installed --no-cache-dir -vvv script_env: