📌  相关文章
📜  火炬设备 - Python 代码示例

📅  最后修改于: 2022-03-11 14:45:51.721000             🧑  作者: Mango

代码示例1
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')