pytorch 命令

 

判断pytorch是否支持cuda

print(torch.cuda.is_available())

 

posted @ 2022-01-03 17:13  小小灰迪  阅读(33)  评论(0)  编辑  收藏  举报