Requirements Linux Python 3.5+ PyTorch 1.11 TensorBoard CUDA 11.0+ GCC 4.9+ NumPy 1.11+ PyYaml Pandas h5py joblib