Skip to content
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
7 changes: 4 additions & 3 deletions recipe/meta.yaml
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
{% set name = "deepspeed" %}
{% set version = "0.11.1" %}
{% set version = "0.11.2" %}
{% set number = 0 %}

{% set torch_proc_type = "cuda" if cuda_compiler_version != "None" else "cpu" %}
Expand All @@ -15,7 +15,7 @@ package:

source:
url: https://pypi.io/packages/source/{{ name[0] }}/{{ name }}/{{ name }}-{{ version }}.tar.gz
sha256: fa342f11f971e1bb002a537adfea6ac4c089b7a4da5aef5f2216a472cb35be7f
sha256: 6d6f716882a3a92ebea37aaeec05a063dc0fe1445d3ab2fc2b603ba87a8a2300

build:
number: {{ number }}
Expand Down Expand Up @@ -66,7 +66,8 @@ requirements:
- psutil
- py-cpuinfo
# PyDantic <2.0.0 is required until DeepSpeed updates to be compatible with newer PyDantic versions.
- pydantic <2.0.0
- pydantic
- pynvml
- python
- pytorch # [cuda_compiler_version in (undefined, 'None')]
- tqdm
Expand Down