花費 64 ms
pytorch中 model.cuda的作用

在pytorch中,即使是有GPU的機器,它也不會自動使用GPU,而是需要在程序中顯示指定。調用model.cuda(),可以將模型加載到GPU上去。這種方法不被提倡,而建議使用model.to(de ...

Wed Oct 28 22:18:00 CST 2020 0 2057

 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM