search

PharMolix / OpenBioMed

MIT License
713 stars 82 forks source link

关于某些版本的cuda无法适配到合适的Pyg依赖包 #36

Open Greay83 opened 1 year ago

Greay83 commented 1 year ago

您好，你们做的工作非常酷，我非常感兴趣。我这边服务器的cuda版本是9.1和11.0，"requirements.txt"里面要求torch>=1.13.0，但是我在下载Pyg的依赖包的页面(https://data.pyg.org/whl/) 没用找到适配的选项，也就是没用匹配的torch+cuda版本，由于服务器是公用的，不太方便直接更改cuda版本，请问还有其他办法能解决这个问题吗？

期待您的回复，谢谢！

icycookies commented 1 year ago

可以尝试torch-1.7.0+cu110和对应的pyg版本，如因版本差异导致运行失败可继续在该issue中回复 :)