torch.load() 分段错误(核心转储)

问题描述 投票:0回答:1

python:3.6.10,火炬:1.2.0,cuda:9.2

当我使用 torch.load() 时,出现分段错误(核心转储),torch 版本是 1.2.0 enter image description here

model_xl = torch.load("./data/models_saved/xl_model.pth")

经过我测试,如果我安装cpu版本的torch,没有问题,但是如果我安装gpu版本的torch,就会出现这个问题,有人可以帮助我吗?

pytorch segmentation-fault
1个回答
0
投票

就我而言,我已经测试了所有版本的nvidia驱动程序,但没有用。 于是上网搜了一下,出来更新BIOS,于是更新了一下,果然成功了。更新BIOS。

https://github.com/pytorch/pytorch/issues/74235

© www.soinside.com 2019 - 2024. All rights reserved.