摘要: 默认已经安装好了pytorch和cuda环境 1、检查pytorch和cuda版本 import torch print(torch.__version__) #pytorch版本 print(torch.version.cuda) #cuda版本 2、以pytorch1.12.*及cuda11.6 阅读全文
posted @ 2022-11-12 10:41 一条土狗 阅读(93) 评论(0) 推荐(0) 编辑