mirror of https://github.com/hpcaitech/ColossalAI
aibig-modeldata-parallelismdeep-learningdistributed-computingfoundation-modelsheterogeneous-traininghpcinferencelarge-scalemodel-parallelismpipeline-parallelism
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
32 lines
662 B
32 lines
662 B
name: ldm |
|
channels: |
|
- pytorch |
|
- defaults |
|
dependencies: |
|
- python=3.9.12 |
|
- pip=20.3 |
|
- cudatoolkit=11.3 |
|
- pytorch=1.12.1 |
|
- torchvision=0.13.1 |
|
- numpy=1.23.1 |
|
- pip: |
|
- albumentations==1.3.0 |
|
- opencv-python==4.6.0.66 |
|
- imageio==2.9.0 |
|
- imageio-ffmpeg==0.4.2 |
|
- omegaconf==2.1.1 |
|
- test-tube>=0.7.5 |
|
- streamlit==1.12.1 |
|
- einops==0.3.0 |
|
- transformers |
|
- webdataset==0.2.5 |
|
- kornia==0.6 |
|
- open_clip_torch==2.0.2 |
|
- invisible-watermark>=0.1.5 |
|
- streamlit-drawable-canvas==0.8.0 |
|
- torchmetrics==0.7.0 |
|
- prefetch_generator |
|
- datasets |
|
- colossalai==0.2.5 |
|
- lightning==1.9.0 |
|
- -e .
|
|
|