检测是否torch的CUDA是否可用的代码

import torch
print(torch.cuda.is_available())
# 返回True 接着用下列代码进一步测试
print(torch.zeros(1).cuda())
posted @ 2022-07-06 17:13  乌蝇哥  阅读(579)  评论(0)  编辑  收藏  举报