Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
requirements.txt 129 Bytes
Newer Older
yuxionghe's avatar
Add files  
yuxionghe committed
1
2
3
4
5 
torch==1.2
torchvision==0.4.0
tqdm
psutil
tensorboardX==1.8
dependabot[bot]'s avatar
Bump tensorflow-gpu from 1.14.0 to 1.15.2 (#39)  
dependabot[bot] committed
6 
tensorflow-gpu==1.15.2
yuxionghe's avatar
Add files  
yuxionghe committed
7 
nvidia-ml-py3
Shaden Smith's avatar
Testing documentation (#10)  
Shaden Smith committed
8 
pytest
Shaden Smith's avatar
Enables NCCL backend in @distributed_test (#13)  
Shaden Smith committed
9 
pytest-forked
yuxionghe's avatar
Add files  
yuxionghe committed
10 
pre-commit