Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
requirements.txt 105 Bytes
Newer Older
Jeff Rasley's avatar
Add back torch/torchvision requirements  
Jeff Rasley committed
1
2 
torch>=1.2
torchvision>=0.4.0
yuxionghe's avatar
Add files  
yuxionghe committed
3
4
5 
tqdm
psutil
tensorboardX==1.8
Shaden Smith's avatar
Testing documentation (#10)  
Shaden Smith committed
6 
pytest
Shaden Smith's avatar
Enables NCCL backend in @distributed_test (#13)  
Shaden Smith committed
7 
pytest-forked
yuxionghe's avatar
Add files  
yuxionghe committed
8 
pre-commit
Jeff Rasley's avatar
Transformer kernel release (#242)  
Jeff Rasley committed
9 
clang-format