解决pycharm中: OSError: [WinError 1455] 页面文件太小,无法完成操作以及GPU调不起来的问题

解决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)

二、截图

在这里插入图片描述

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值