2022年7月12日
pytorch 命令
摘要： 1.检查cuda import torch torch.cuda.is_available() # 检查cuda是否可用 torch.version.cuda # 查看cuda版本 2.检查cudnn torch.backends.cudnn.is_available() # 检查cudnn是否可用 阅读全文
posted @ 2022-07-12 18:57 无左无右 阅读(35) 评论(0) 推荐(0) 编辑