今天跑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小时内删除。
留言与评论(共有 0 条评论) |