pytorch 預設使用從0開始的gpu,如果gpu0正在執行程式,需要指定其他gpu。 有如下兩種方法來指定需要使用的gpu。 1. 類似tensorflow指定gpu的方式, ...
確定! 回上一頁