2025年10月25日
Windows系统安装了CUDA，但是Python的PyTorch还是没用gpu进行训练
摘要： 本文基于 PyTorch训练框架 电脑装了CUDA 但是python训练的时候显示的还是用了cpu进行训练 def print_device_info(): """打印设备信息""" print("PyTorch版本:", torch.__version__) print("CUDA是否可用:", 阅读全文
posted @ 2025-10-25 01:39 yvioo 阅读(4) 评论(0) 推荐(0)