PyTorch 中cuda 相关操作的官方教程:网址. torch.cuda.is_available() :返回bool 类型;GPU 是否可用. torch.device(device) : 使用GPU 或者CPU ...
確定! 回上一頁