mirrors
/
PyRetri
mirror of https://github.com/PyRetri/PyRetri.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

upload

pull/5/head
hubenyi 2020-04-29 11:58:31 +08:00
parent 2d42f06c44
commit 8b924bb59e
1 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
docs/INSTALL.md
View File

@ -3,10 +3,9 @@ Installation
## Requirements
- Linux (Windows is not officially supported)
- Python 3.5 or higher
- Python 3
- PyTorch 1.2.0 or higher
- torchvison 0.4.0 or higher
- CUDA 9.0 or higher
- numpy
- sklearn
- yacs
@ -14,9 +13,9 @@ Installation
Our experiments are conducted on the following environment:
- Python 3.5
- PyTorch 1.2.0
- torchvision 0.4.0
- CUDA 9.0
- numpy 1.17.2
- sklearn 0.21.3
- tqdm 4.36.1