@lgvaz you have a 1070 GPU, but you must’ve installed pytorch for CUDA 7.5. What it is doing is recompiling pytorch for your GPU (cuda8 supported GPU).
If you reinstall the CUDA8 version of pytorch, the freezing wont happen. Select the right option in our version selector on http://pytorch.org 120