CPU 不支持半精度计算,重新安装成GPU版本的pytorch 就可以 查看安装的pytorch是否能使用CUDA进行加速计算 import torch torch.cuda.is_available()