GPU CPU使用问题 - Githubissues

troublemaker-r / Chinese_Coreference_Resolution

基于SpanBert的中文指代消解，pytorch实现

95 stars 20 forks source link

Open YXD942 opened 3 years ago

YXD942 commented 3 years ago

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

troublemaker-r commented 3 years ago

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

troublemaker-r commented 3 years ago

@YXD942

YXD942 commented 3 years ago

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

YXD942 commented 3 years ago

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

troublemaker-r commented 3 years ago

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

cuikai-ai commented 1 year ago

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

Jeru2023 commented 8 months ago

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

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