PyTorch使用问题汇总

1. 找不到RandomReSizedCrop

module 'torchvision.transforms' has no attribute 'RandomReSizedCrop'

原因：

在pytorch的老版本中该类叫RandomReSizedCrop，在新版本中叫RandomResizedCrop，二者只差一个字母大小写，不易发现。

2. 没有训练任务，显存却被占用

解决方法：

安装工具

sudo apt-get install psmisc

运行

fuser -v /dev/nvidia*

查看占用显存的进程的pid，然后杀死

kill -9 [PID]