File size: 542 Bytes
73c83cf |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 |
name: idm
channels:
- pytorch
- nvidia
- defaults
dependencies:
- python=3.10.0=h12debd9_5
- pytorch=2.0.1=py3.10_cuda11.8_cudnn8.7.0_0
- pytorch-cuda=11.8=h7e8668a_5
- torchaudio=2.0.2=py310_cu118
- torchtriton=2.0.0=py310
- torchvision=0.15.2=py310_cu118
- pip=23.3.1=py310h06a4308_0
- pip:
- accelerate==0.25.0
- torchmetrics==1.2.1
- tqdm==4.66.1
- transformers==4.36.2
- diffusers==0.25.0
- einops==0.7.0
- bitsandbytes==0.39.0
- scipy==1.11.1
- opencv-python
|