我是在import dgl时报的错,conda list了一下,发现dgl是cuda10.2的,但是cudatoolkit是10.1的,重新安装一个10.2的cudatoolkit就好了

!conda install cudatoolkit=10.2 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/linux-64/

或者装个10.1的dgl-cu10.1

!conda install -c dglteam dgl-cuda10.1==0.4.3post2
Logo

欢迎来到由智源人工智能研究院发起的Triton中文社区,这里是一个汇聚了AI开发者、数据科学家、机器学习爱好者以及业界专家的活力平台。我们致力于成为业内领先的Triton技术交流与应用分享的殿堂,为推动人工智能技术的普及与深化应用贡献力量。

更多推荐