suragnair / alpha-zero-general

A clean implementation based on AlphaZero for any game in any framework + tutorial + Othello/Gobang/TicTacToe/Connect4 and more
MIT License
3.9k stars 1.04k forks source link

Minimizing requirements.txt file #217

Open mikhail opened 4 years ago

mikhail commented 4 years ago

I started with a blank requirements file and added libraries 1 by 1. For my game the resulting file was significantly smaller:

coloredlogs==14.0
torch==1.6.0
tqdm==4.48.2

Plus pyyaml & pymunk needed by my own project.

For comparison here's the current requirements file in the docker directory:

cffi==1.11.5
coloredlogs==14.0
cython==0.28.3
flask==1.0
gitpython==2.1.11
matplotlib==2.1.1
numpy==1.14.5
pandas==0.23.1
scipy==1.1.0
scikit-learn==0.19.1
scikit-image==0.14.0
torchfile==0.1.0
torchvision==0.2.1
tqdm==4.19.5
visdom==0.1.7

What are all the other libraries in the requirements file? Should we create separate files for pytorch/tensorflow etc, and perhaps for different versions of python?

suragnair commented 4 years ago

Yeah that would be great!