requirements.txt

gym
Pillow
docopt
PyYAML
tensorboard
moviepy
structlog
networkx
scipy
jsonlines
openpyxl
# dvc
# boto3
pyarrow
# torch_geometric
# torch==1.7.1
# torchvision

# python -c "import torch; print(torch.__version__)"
# python -c "import torch; print(torch.version.cuda)"
# pip install torch-scatter==latest+cu101 -f https://pytorch-geometric.com/whl/torch-1.7.0.html
# pip install torch-sparse==latest+cu101 -f https://pytorch-geometric.com/whl/torch-1.7.0.html
# pip install torch-cluster==latest+cu101 -f https://pytorch-geometric.com/whl/torch-1.7.0.html
# pip install torch-spline-conv==latest+cu101 -f https://pytorch-geometric.com/whl/torch-1.7.0.html
# pip install torch-geometric

# pip install torch-scatter==latest+cu102 -f https://pytorch-geometric.com/whl/torch-1.7.1.html
# pip install torch-sparse==latest+cu102 -f https://pytorch-geometric.com/whl/torch-1.7.1.html
# pip install torch-cluster==latest+cu102 -f https://pytorch-geometric.com/whl/torch-1.7.1.html
# pip install torch-spline-conv==latest+cu102 -f https://pytorch-geometric.com/whl/torch-1.7.1.html
# pip install torch-geometric