System: Using docker image pytorch/pytorch:1.7.1-cuda11.0-cudnn8-runtime
apt-get update && apt install -y libsm6 libxext6

python packages:
pip install -r requirement.txt
