在pytorch中为Module和Tensor指定GPU的例子 pytorch指定GPU在用pytorch写CNN的时候，发现一运行程序就卡住，然后cpu占用率100%，nvidia-smi查 指定GPU