训练选择显卡设备

[wwwww-k]

请问如何调整训练的显卡，比如说想用第二张卡训练这种，或者选择多卡训练

[JlexZhong]

请问您解决了吗

[wwwww-k]

请问您解决了吗

没有解决

[sustech-lz]

请问您解决了吗

想请问一下，如何才能自行设置显卡呢？有的时候需要用的是别的显卡训练，但是模型默认会从0号开始使用

[wwwww-k]

没解决，那个项目后面没在跑了

…

------------------ 原始邮件 ------------------ 发件人: "HKUDS/GraphGPT" ***@***.***>; 发送时间: 2024年10月25日(星期五) 晚上8:17 ***@***.***>; ***@***.******@***.***>; 主题: Re: [HKUDS/GraphGPT] 训练选择显卡设备 (Issue #74) 请问您解决了吗 想请问一下，如何才能自行设置显卡呢？有的时候需要用的是别的显卡训练，但是模型默认会从0号开始使用 — Reply to this email directly, view it on GitHub, or unsubscribe. You are receiving this because you authored the thread.Message ID: ***@***.***>

[sustech-lz]

没解决，那个项目后面没在跑了
…
------------------ 原始邮件 ------------------ 发件人: "HKUDS/GraphGPT" @.>; 发送时间: 2024年10月25日(星期五) 晚上8:17 @.>; @.@.>; 主题: Re: [HKUDS/GraphGPT] 训练选择显卡设备 (Issue #74) 请问您解决了吗 想请问一下，如何才能自行设置显卡呢？有的时候需要用的是别的显卡训练，但是模型默认会从0号开始使用 — Reply to this email directly, view it on GitHub, or unsubscribe. You are receiving this because you authored the thread.Message ID: @.***>

CUDA_VISIBLE_DEVICES=4,5 我在文件当中设置这个就行了，gpu个数可以在train里面看到，但是我发现多gpu跑的速度不如2个gpu很奇怪