Ptt 大爆卦 | PyTorch CPU,GPU - 前往 https://juejin.cn/s/%E6%80%8E%E4%B9%88%E7%9C%8Bpytorch%E6%98%AFcpu%E8%BF%98%E6%98%AFGPU

你即將離開本站

並前往https://juejin.cn/s/%E6%80%8E%E4%B9%88%E7%9C%8Bpytorch%E6%98%AFcpu%E8%BF%98%E6%98%AFGPU

怎么看pytorch是cpu还是GPU - 稀土掘金

可以使用PyTorch的torch.cuda.is_available()函数来检测是否有可用的GPU。如果返回True，则表示有可用的GPU；如果返回False，则表示当前使用的是CPU。可以这样使用：

確定！回上一頁

查詢「PyTorch CPU,GPU」的人也找了：