2022年4月12日
验证pytorch是否安装成功
摘要： import torch #查看torch版本 print(torch.version) #查看cuda版本 print(torch.version.cuda) #GPU是否可用 print(torch.cuda.is_available()) #返回gpu数量 print(torch.cuda.d 阅读全文
posted @ 2022-04-12 11:15 Touming 阅读(6175) 评论(0) 推荐(0)