Chinese_Coreference_Resolution icon indicating copy to clipboard operation
Chinese_Coreference_Resolution copied to clipboard

Published 20 hours ago • verified publisher icon troublemaker-r

GPU CPU使用问题

Open YXD942 opened this issue 3 years ago • 7 comments

请问我训练的时候发现机器使用了CPU来进行训练，我的GPU是8GB的，原因是因为GPU大小不够吗？还是说需要手动配置呢

YXD942 avatar Aug 13 '21 09:08 YXD942

首先，检查是否安装了cup版；其次，是否用了该环境跑；最后，可以手动设置使用GPU。一般不会跑不动，跑不动会报显存错误。

troublemaker-r avatar Aug 17 '21 08:08 troublemaker-r

@YXD942

troublemaker-r avatar Aug 17 '21 08:08 troublemaker-r

谢谢，问题已经解决，现在正在等待训练结果

YXD942 avatar Aug 17 '21 08:08 YXD942

请问如果要根据需求训练新的模型，数据的格式是按照data里的格式进行处理吗？需要自行写脚本处理吗 @troublemaker-r

YXD942 avatar Aug 17 '21 09:08 YXD942

@YXD942 看你情况，如果改成他的格式方便的话，就和它一样就好。你的数据格式不同的话当然需要写数据处理脚本，可能还比较麻烦的哦

troublemaker-r avatar Aug 17 '21 09:08 troublemaker-r

image 遇到以下问题，请教一下怎么解决？单cpu单卡，demo里os.environ["CUDA_VISIBLE_DEVICES"] = “2”会调用cpu跑，改成0就会报这个错

cuikai-ai avatar Apr 09 '23 02:04 cuikai-ai

image 遇到以下问题，请教一下怎么解决？单cpu单卡，demo里os.environ["CUDA_VISIBLE_DEVICES"] = “2”会调用cpu跑，改成0就会报这个错

把那两行环境变量设置的代码都注释掉，然后确保所有变量都在cuda device计算

Jeru2023 avatar Mar 12 '24 11:03 Jeru2023