DeepSpeed

1
channels:
2
  - nvidia/label/cuda-11.8.0
3
  - pytorch # or pytorch-nightly
4
  - conda-forge
5
dependencies:
6
  - pytorch
7
  - torchvision
8
  - torchaudio
9
  - cuda
10
  - pytorch-cuda=11.8
11
  - compilers
12
  - sysroot_linux-64==2.17
13
  - gcc==11.4
14
  - ninja
15
  - py-cpuinfo
16
  - libaio
17
  - ca-certificates
18
  - certifi
19
  - openssl
20
  - python=3.10
21
  - pydantic
22