当前位置: 主页 > 图像 >

undefined symbol: _ZN2at19UndefinedTensorImpl10_singletonE

时间:2022-08-11  作者:haden   点击:
【摘要】在Colab尝试复现一篇神经网络论文的时候,遇到了报错:```ImportError: /.../_C.cpython-36m-x86_64-linux-gnu.so: undefined symbol: _ZN2at19UndefinedTensorImpl10_singletonE```查找资料发现应该是cuda、torch、torchvi

在Colab尝试复现一篇神经网络论文的时候,遇到了报错:

ImportError: /.../_C.cpython-36m-x86_64-linux-gnu.so: undefined symbol: _ZN2at19UndefinedTensorImpl10_singletonE

查找资料发现应该是cuda、torch、torchvision版本不对应。

查看torch的cuda版本

python
>>> import torch
>>> torch.version.cuda
'10.0.130'

查看机器的cuda版本

nvcc -V
...
Cuda compilation tools, release 9.0, V9.0.176

发现并不对应,去官网找到符合自己要求的版本https://pytorch.org/get-started/previous-versions/

按照要求重新按照:

!pip install torch==1.8.0+cu111 torchvision==0.9.0+cu111 torchaudio==0.8.0 -f https://download.pytorch.org/whl/torch_stable.html
顶一下
(0)
0%
踩一下
(0)
0%
发表评论
请自觉遵守互联网相关的政策法规,严禁发布色情、暴力、反动的言论。
评价:
验证码: 点击我更换图片