老师，我的pytorch=1.4,torchvision=0.5,在训练的时候CPU占用率并没有超过70%，可是运行错误，这是怎么回事？

首页课程实战体系课手记专栏慕课教程

老师，我的pytorch=1.4,torchvision=0.5,在训练的时候CPU占用率并没有超过70%，可是运行错误，这是怎么回事？

来源：8-25 训练脚本编写及演示

AS小黑

2020-04-19

图片描述

写回答

1回答

胖虎

2020-04-19

已采纳

默认是GPU训练的啊，你这个报错是oom了，减小batchsize试试

0

2

胖虎

回复

AS小黑

没事的，应该得

2020-04-20

共2条回复

基于Pytorch热门深度学习框架从零开发NLP聊天机器人

基于中科院所内项目，Seq2Seq+attention主流算法，实战聊天机器人

273 学习 · 107 问题

相似问题

这是什么原因？

回答 2

老师，最终的代码为什么都执行不了，代码里面有好多错误，能重新提供一份么

回答 1

训练脚本运行报错 out of range

回答 2

老师，我把train.py文件的数据修改如下，训练并保存完后，运行test.py出现了除以0的问题，这是怎么回事？

回答 2

导入库就给我报错RuntimeError: builtin cannot be used as a value:，

回答 1

打开慕课网App查看更多内容