【Pytorch】使用过程中的一些报错记录

 

ImportError: DLL load failed: 页面文件太小，无法完成操作

python代码训练神经网络时，报错：“ImportError: DLL load failed: 页面文件太小，无法完成操作。”

主要是公司电脑是windows，有些在本地上调和集群上环境不一样。

出现这个问题可能有两种情况

1、不止在运行一个项目，另一个项目的python程序也在运行，关掉就可以了。
2、windows操作系统不支持python的多进程操作。而神经网络用到多进程的地方在数据集加载上，所以将DataLoader中的参数num_workers设置为0即可。

    train_loader = torch.utils.data.DataLoader(
        train_dataset,
        batch_size=16,
        shuffle=True,
        num_workers=0,  # 设置为0
    )