关于SSLRec运行时出现“Cuda Kernel Error”？

HKUDS / SSLRec

[WSDM'2024 Oral] "SSLRec: A Self-Supervised Learning Framework for Recommendation"

https://arxiv.org/abs/2308.05697

Apache License 2.0

467 stars 60 forks source link

Closed Chuckie-XC1028 closed 6 months ago

Chuckie-XC1028 commented 6 months ago

问题描述：直接运行SSLRec中的LightGCN代码，调试到EdgeDrop是出现“Cuda Kernel Error”，作者有知道原因吗？如下是报错图片以及GPU使用情况：
测试环境描述：
- torch==1.11.0
- scipy==1.11.4
- numpy==1.26.4

Chuckie-XC1028 commented 6 months ago

更新：上述问题出现在AutoDL云平台中的2080ti性能卡中，但使用3090性能卡时又不会出现此问题，这让我很疑惑🤣? 问题：请问贵团队是否在其它性能卡中测试过SSLrec代码呢？可以帮忙解惑一下嘛，不胜感激🙏。

Re-bin commented 6 months ago

嗨 👋！

感谢关注SSLRec！我们测试环境确实用的是RTX 3090。

不过一般这种报错是出现了访问越界，看上去28行好像是单独加上去的，原始的edgedrop的代码里面没有这一行代码，会不会注释之后，或许可以正常运行。

希望能帮到你 :)

Best regards, Xubin

Chuckie-XC1028 commented 6 months ago

非常感谢Re-bin的回答👏，这种现象看上去是随机性的(或者说是性能卡本身原因)，刚测试AutoDL云平台中其他2080ti性能卡时，该现象不在存在😂.

Re-bin commented 6 months ago

好的，感谢您对SSLRec的关注！如果有后续有改进意见和改进代码，欢迎提issue和pr :)

Best regards, Xubin