deepspeed
/
environment.yml
21 строка · 340.0 Байт
1channels:
2- nvidia/label/cuda-11.8.0
3- pytorch # or pytorch-nightly
4- conda-forge
5dependencies:
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