解决pycharm中: OSError: [WinError 1455] 页面文件太小,无法完成操作以及GPU调不起来的问题
提示:相信大家都已经看过关于修改虚拟内存解决问题的方式,这里我就不对此再赘述了,主要是介绍另外一种方式。
使用的是否是pytorch项目,如果是,找到模型训练的train.py文件,在文件中找到 parser.add_argument,寻找关于num_worker或者(num_workers)的设置,将num_worker设为0。
解释:num_worker代表训练数据生成器线程数,如果你指定GPU负责训练模型,设为0之后就会调起GPU训练,否则你会发现GPU一直调不起来,一直是CPU在训练
pytorch中指定GPU训练代码
use_cuda = torch.cuda.is_available()
device = torch.device("cuda:0" if use_cuda else "cpu")
model_ = model_.to(device)
二、截图
。