如何解决pytorch报错:AssertionError: Invalid device id的问题：https://www.yisu.com/zixun/152315.html
TypeError: can’t convert CUDA tensor to numpy. Use Tensor.cpu() to copy the tensor to host memory first. ：https://www.cnblogs.com/xiaodai0/p/10677363.html