怎么指定gpu运行？怎么调整batchsize？
#13862
Replies: 1 comment 2 replies
-
Beta Was this translation helpful? Give feedback.
2 replies
-
|
您好，没找到相关信息指定gpu进行训练 |
Beta Was this translation helpful? Give feedback.
-
|
在执行训练推理时添加下面代码即可 CUDA_VISIBLE_DEVICES=(GPU id) python train.py |
Beta Was this translation helpful? Give feedback.
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
-
怎么指定gpu运行？怎么调整batchsize？
Beta Was this translation helpful? Give feedback.
All reactions