Recommended version of all package for this repo torch 1.4 torchvision 0.5.0 cuda 10.1 gcc 5.5.0 g++ 5.5.0