pytorch 命令

 

判断pytorch是否支持cuda

print(torch.cuda.is_available())

 

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