“sum_cpu“ not implemented for ‘Bool‘

最新推荐文章于 2024-09-30 23:35:28 发布

喝粥也会胖的唐僧

最新推荐文章于 2024-09-30 23:35:28 发布

阅读量2.1k

点赞数

分类专栏：图论 Python

本文链接：https://blog.csdn.net/zhou_438/article/details/108710688

版权

Python 同时被 2 个专栏收录

126 篇文章

订阅专栏

图论

2 篇文章

订阅专栏

本文解决了一个在使用DGL进行图神经网络训练过程中遇到的问题：由于使用了PyTorch CPU版而导致的错误。文章详细介绍了如何正确安装GPU版本的PyTorch，并给出了具体的conda安装命令。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

在使用dgl训练图神经网络的时候报错了：

"sum_cpu" not implemented for 'Bool'

原因是dgl只支持gpu版，而安装的 pytorch是安装是的cpu版，解决方法是重新安装pytoch为gpu版

conda install pytorch==1.2.0 torchvision==0.4.0 cudatoolkit=10.0 -c pytorch

注意的是：因为自己机器上是cuda10，所以安装的是稍低一些的版本，反正pytorch1.5及其以上是不支持10.0的

不然会出错：

ERROR: Could not find a version that satisfies the requirement torch==1.5.0+cu100 (from versions: 0.1.2, 0.1.2.post1, 0.1.2.post2, 0.4.1, 0.4.1.post2, 1.0.0, 1.0.1, 1.0.1.post2, 1.1.0, 1.2.0, 1.2.0+cpu, 1.2.0+cu92, 1.3.0, 1.3.0+cpu, 1.3.0+cu100, 1.3.0+cu92, 1.3.1, 1.3.1+cpu, 1.3.1+cu100, 1.3.1+cu92, 1.4.0, 1.4.0+cpu, 1.4.0+cu100, 1.4.0+cu92, 1.5.0, 1.5.0+cpu, 1.5.0+cu101, 1.5.0+cu92, 1.5.1, 1.5.1+cpu, 1.5.1+cu101, 1.5.1+cu92, 1.6.0, 1.6.0+cpu, 1.6.0+cu101, 1.6.0+cu92)
ERROR: No matching distribution found for torch==1.5.0+cu100