- 博客(7)
- 收藏
- 关注
RSS订阅原创 pytorch模型保存的方法以及区别
运行torch.save(model,PATH)保存整个模型，包括其结构和参数，加载时无需重新定义模型结构，不过可能导致兼容性问题，特别是不同版本的PyTorch之间。运行torch.save(model.state_dict(),PATH)可在训练过程中同时保存模型参数和优化器状态，便于恢复。加载保存的字典，然后分别加载模型和优化器的状态。3、保存模型和优化器的状态。
2025-03-18 19:36:33
128
原创 分布式训练教程（多节点多显卡）
使用(DDP) 包裹模型，实现多节点多 GPU 训练。节点1存在2张显卡，可利用利用(DP) 或(DDP) 来并行处理数据。使用 DataParallel (DP)model = nn.DataParallel(model, device_ids=[0, 1]) # 使用两张显卡使用 DistributedDataParallel (DDP)在节点 2 和 3 上，由于只有一张显卡，直接使用。
2025-03-18 15:52:22
491
原创 记录：虚拟环境中安装GPU版pytorch
之前装过一次gpu版pytorch，后来发现仅仅在base环境中的torch.cuda.is_available()=true，另外几个虚拟环境中torch.cuda.is_available()=false，然而我一般使用的环境并不是base……
2023-02-08 10:37:26
486
原创 解决Could not load dynamic library ‘cudart64_101.dll’； dlerror: cudart64_101.dll not found
解决Could not load dynamic library ‘cudart64_101.dll’; dlerror: cudart64_101.dll not found
2022-08-12 17:31:32
6156
2
原创 python离线安装apex，无NVIDIA版本
问题：Warning: Torch did not find available GPUs on this system.If your intention is to cross-compile, this is not an error.By default, Apex will cross-compile for Pascal (compute capabilities 6.0, 6.1, 6.2),Volta (compute capability 7.0), Turing (compute
2022-07-09 13:28:20
4086
21
原创 powershell执行策略报错
无法加载文件 X:\x\x，因为在此系统上禁止运行脚本。有关详细信息，请参阅 https:/go.microsoft.com/fwlink/?LinkID=135170 中的 about_Execution_Policies。
2022-01-08 11:33:21
672
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人