Multi gpu usage in pytorch for faster inference. ... when gpu id is not mentioned inside the function cuda(), model is put into the default gpu.
確定! 回上一頁