pytorch错误汇总

蔚落 2023-03-13 05:26 1阅读 0赞

*  [pytorch与cuda版本的对应关系][pytorch_cuda]
 *  undefined symbol: \_\_cudaPopCallConfiguration。   导致该问题的原因为系统cuda版本与torch编译是的cuda版本不一致。

python -c 'import torch; print (torch.version.cuda)' #查看当前torch编译的cuda版本
    nvcc --version  #查看系统的cuda版本

*  版本不一致时，可安装对应当前机器cuda版本的pytorch版本即可。若是github上给定版本的pytorch，此时就需要安装对应torch版本的cuda了。先在[nvidia官网][nvidia]找到对应版本的cuda，根据提示选择适合系统的较大的那个文件。

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2h6aGoyMDA3_size_16_color_FFFFFF_t_70][]

linux安装时，driver不安装，其他选择yes. 然后修改/usr/local/cuda的软链接到新安装的cuda。

rm /usr/local/cuda  #删除旧的软链接
    ln -s  /usr/local/cuda-9.0/ /usr/local/cuda #创建新的软连接

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2h6aGoyMDA3_size_16_color_FFFFFF_t_70 1][]

参考文献：

1.  [ImportError: undefined symbol: \_\_cudaPopCallConfiguration][ImportError_ undefined symbol_ _cudaPopCallConfiguration]
2.  [安装多版本 cuda ，多版本之间切换][cuda]

[pytorch_cuda]: https://pytorch.org/get-started/previous-versions/
[nvidia]: https://developer.nvidia.com/cuda-toolkit-archive
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2h6aGoyMDA3_size_16_color_FFFFFF_t_70]: /images/20230312/9056f4d80d45417b9a74cdf00756011f.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2h6aGoyMDA3_size_16_color_FFFFFF_t_70 1]: /images/20230312/9936e0e55c8f4e4088abe414176412bb.png
[ImportError_ undefined symbol_ _cudaPopCallConfiguration]: https://github.com/rusty1s/pytorch_scatter/issues/19
[cuda]: https://blog.csdn.net/Maple2014/article/details/78574275