- 博客(2)
- 收藏
- 关注
RSS订阅原创 pytroch加载gpu训练的DataParallel模型
pytroch加载gpu训练的DataParallel模型加载.pkl模型到cpu上model = torch.load('xxx.pkl')model.cpu()报错，cpu环境下不能直接导入gpu训练的DataParallel模型，找了一圈，有建议遍历网络把module去掉的。找到一个简单方法，在gpu上把模型转化掉model = torch.load('xxx.pkl')real_model = model.moduletorch.save(real_model, 'xxxx.pkl'
2020-09-25 16:10:07
131
原创 RuntimeError: cuda runtime error (10) : invalid device ordinal at /pytorch/torch/csrc/cuda/Module.cp
RuntimeError: cuda runtime error (10) : invalid device ordinal at /pytorch/torch/csrc/cuda/Module.cpp:59服务器有四块卡，load之前训练的模型，由于第一块GPU已经使用，使用第四块gpu，指令之前加了CUDA_VISIBLE_DEVICES=3指定GPU的id。出现以下错误：查了一些资料，在import pytorch之前插入以下import osos.environ["CUDA_VISIBL
2020-09-25 10:00:51
577
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人