how to change GPU number

Alexander-H-Liu / End-to-end-ASR-Pytorch

This is an open source project (formerly named Listen, Attend and Spell - PyTorch Implementation) for end-to-end ASR implemented with Pytorch, the well known deep learning toolkit.

MIT License

1.18k stars 317 forks source link

how to change GPU number #78

Open diveknown opened 2 years ago

diveknown commented 2 years ago

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

vectominist commented 2 years ago

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.