End-to-end-ASR-Pytorch icon indicating copy to clipboard operation
End-to-end-ASR-Pytorch copied to clipboard

Published 20 hours ago • verified publisher icon Alexander-H-Liu

how to change GPU number

Open diveknown opened this issue 2 years ago • 1 comments

Our lab's GPU0 is used, and I can only use GPU1. It shows CUDA out of memory when using GPU0

diveknown avatar Nov 23 '21 14:11 diveknown

Perhaps you can add CUDA_VISIBLE_DEVICES=1 before the python script when executing, where the number 1 can be changed to your GPU's ID.

vectominist avatar Nov 24 '21 05:11 vectominist