graph4ai / graph4nlp

Graph4nlp is the library for the easy use of Graph Neural Networks for NLP. Welcome to visit our DLG4NLP website (https://dlg4nlp.github.io/index.html) for various learning resources!
Apache License 2.0
1.67k stars 201 forks source link

请问该库包支持CUDA11.2版本嘛 #562

Closed yingdajun closed 2 years ago

yingdajun commented 2 years ago

我的CUDA版本是11.2,请问该库包可以有CUDA11.2版本嘛?

如果我安装CUDA11.0版本,可以嘛

cminus01 commented 2 years ago

For the CUDA 11.0, you can install our library by using 'pip install graph4nlp-cu110 ' or follow the instruction of installation via source code. And you can refer to installation via source code for CUDA 11.2 as well.