1、如下图所示，数据加载时：
这个是transform有问题，我遇到了以下情况：
（1）随机裁剪的尺度大于了原图，
（2）PIL读取的图片再次转为PIL格式。
还有可能是图片路径问题：可以调试查看原因。
2、CUDNN_STATUS_NOT_INITIALIZED
（1）batch_size太大，显存不够。
（2） 检查GPU服务器的CUDA版本，一定要在Pytorch官网安装对应你CUDA版本的Pytorch。
（3） 若提示loss不能反传，则检查分类类别数量是否与实际数据的类别数量一致。
版权声明：本文为weixin_42726887原创文章，遵循CC 4.0 BY-SA版权协议，转载请附上原文出处链接和本声明。