Deep learning CUDA错误:遇到非法内存访问

Deep learning CUDA错误:遇到非法内存访问,deep-learning,pytorch,google-colaboratory,Deep Learning,Pytorch,Google Colaboratory,我无法在Google Colab上对PyTorch张量进行设备转换。我附上下面的代码片段: 请帮帮我

我无法在Google Colab上对PyTorch张量进行设备转换。我附上下面的代码片段:
请帮帮我