复现IGN时遇到的关于dgl和多进程的坑

阅读: 评论:0

复现IGN时遇到的关于dgl和多进程的坑

复现IGN时遇到的关于dgl和多进程的坑

今天跑InteractionGraphNet (IGN)的代码,因为原作repo的环境版本实在太老,pytorch1.3.1+cu92在cu11.4的机器上实在跑不起来,python3.6.5的multiprocessing还有bug,报错`ValueError: signal number 32 out of range.` 要更新到python3.6.6就能解决,但是直接更新python环境就乱了。。

于是构建了一个新版的环境:

conda create -n ign -c pytorch -c rdkit -c dglteam -c anaconda -c conda-forge pytorch==1.8 cudatoolkit=11.1 dgl dgllife scikit-learn scipy tqdm tensorboardx nb_conda jedi==0.17.2 python=3.7 dgl-cuda11.1 rdkit -y

修改几处API就能跑通了,构建graph也可以多进程了。虽然dgl一直报 DGLWarning: Recommend creating graphs by `aph(data)` instead of `dgl.DGLGraph(data)`. 属于旧版dgl430的遗留物,但这种没有任何杀伤力的Warning咱向来是不管的

本文发布于:2024-01-28 20:17:03,感谢您对本站的认可!

本文链接:https://www.4u4v.net/it/170644422610008.html

版权声明:本站内容均来自互联网,仅供演示用,请勿用于商业和其他非法用途。如果侵犯了您的权益请与我们联系,我们将在24小时内删除。

标签:进程   IGN   dgl
留言与评论(共有 0 条评论)
   
验证码:

Copyright ©2019-2022 Comsenz Inc.Powered by ©

网站地图1 网站地图2 网站地图3 网站地图4 网站地图5 网站地图6 网站地图7 网站地图8 网站地图9 网站地图10 网站地图11 网站地图12 网站地图13 网站地图14 网站地图15 网站地图16 网站地图17 网站地图18 网站地图19 网站地图20 网站地图21 网站地图22/a> 网站地图23