Pytorch 1.01 Docker build and trick