ColossalAI/examples/images/vit/requirements.txt

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
[example] add example requirement (#2345) 2 years ago			`colossalai >= 0.1.12`
			`torch >= 1.8.1`
[example] update vit ci script (#2469) * [example] update vit ci script * [example] update requirements * [example] update requirements 2 years ago			`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`