【发布时间】:2021-03-25 13:13:05
【问题描述】:
所以,我正在运行这个深度神经网络。通常,当我使用没有 cuda 的 pytorch 库使用 cpu 训练它时,它运行良好。
但是,我注意到当我安装 pytorch+cuda 11.1 时,每当我尝试枚举 Dataloader 时,都会抛出以下错误:
OSError: [WinError 1455] The paging file is too small for this operation to complete. Error loading "C:\Users\Name\Desktop\Folder\lib\site-packages\torch\lib\cudnn_adv_train64_8.dll" or one of its dependencies.
但是,当我使用 pip 卸载 pytorch+cuda 并在没有 cuda 的情况下安装 pytorch 时,我的脚本运行良好。
pytorch 的两个版本都是 1.8.0
有人知道如何解决这个问题吗?希望能够使用这个 gpu。
注意:这是我设置数据加载器的方式:
train_dataloader = DataLoader(
train_dataset,
batch_size=args.batch_size,
num_workers=args.num_workers,
shuffle=True,
pin_memory=(device == "cuda"),
)
【问题讨论】: