This work was tested with PyTorch 1.2.0, CUDA 9.0, python 3.6 and Ubuntu 16.04. requirements can be found in the file environmentPytorch12.yml. The command to create the environment from the file is: ...