mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2026-04-25 01:03:35 +00:00
8e85d2440a
* [example] update vit ci script * [example] update requirements * [example] update requirements
9 lines
209 B
Plaintext
9 lines
209 B
Plaintext
colossalai >= 0.1.12
|
|
torch >= 1.8.1
|
|
numpy>=1.24.1
|
|
timm>=0.6.12
|
|
titans>=0.0.7
|
|
tqdm>=4.61.2
|
|
transformers>=4.25.1
|
|
nvidia-dali-cuda110>=1.8.0 --extra-index-url https://developer.download.nvidia.com/compute/redist
|