2022年6月22日
Python-将模型和张量加载到GPU
摘要： 在配置好GPU版本的pytorch的环境后，可以先检查一下cuda是否可用 import torch device=torch.device("cuda:0" if torch.cuda.is_available() else "cpu") print(device) 或者是 torch.cuda. 阅读全文
posted @ 2022-06-22 15:37 Khru 阅读(616) 评论(0) 推荐(1) 编辑