This is my setup:

PyTorch Version (e.g., 1.0):  Caffer2 tag v0.4.0

OS (e.g., Linux):  Ubuntu 16.04

How you installed PyTorch (conda, pip, source):  Build from source (tag v0.4.0)

Build command you used (if compiling from source):
Python version:  Python 2.7

CUDA/cuDNN version:  8.0/7.0.5
GPU models and configuration:  GTX 1050