前言,在pytorch中,当服务器上的gpu被占用时,很多时候我们想先用cpu调试下代码,那么就需要进行gpu和cpu的切换。方法1:x.to(device)把device 作为一个可变参数, ...
確定! 回上一頁