ChatGLM-Tuning icon indicating copy to clipboard operation
ChatGLM-Tuning copied to clipboard

Published 20 hours ago • verified publisher icon mymusise

如何指定GPU进行训练？

Open dsh54054 opened this issue 1 year ago • 2 comments

代码跑起来模型第一块GPU进行训练，如何指定GPU训练呢

dsh54054 avatar Mar 23 '23 10:03 dsh54054

CUDA_VISIBLE_DEVICES=gpu_id python your_script.py

superhg avatar Mar 23 '23 10:03 superhg

CUDA_VISIBLE_DEVICES

大佬，这种方式可行吗

tjulh avatar Jun 28 '23 02:06 tjulh