前言,在pytorch中,當服務器上的gpu被占用時,很多時候我們想先用cpu調試下代碼, ... import torch print(torch.cuda.device_count()) # 可用gpu ...
確定! 回上一頁