mirror of
https://github.com/deepspeedai/DeepSpeed.git
synced 2025-10-20 15:33:51 +08:00
22 lines
340 B
YAML
22 lines
340 B
YAML
channels:
|
|
- nvidia/label/cuda-11.8.0
|
|
- pytorch # or pytorch-nightly
|
|
- conda-forge
|
|
dependencies:
|
|
- pytorch
|
|
- torchvision
|
|
- torchaudio
|
|
- cuda
|
|
- pytorch-cuda=11.8
|
|
- compilers
|
|
- sysroot_linux-64==2.17
|
|
- gcc==11.4
|
|
- ninja
|
|
- py-cpuinfo
|
|
- libaio
|
|
- ca-certificates
|
|
- certifi
|
|
- openssl
|
|
- python=3.10
|
|
- pydantic
|