“sum_cpu“ not implemented for ‘Bool‘

在使用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

猜你喜欢

转载自blog.csdn.net/zhou_438/article/details/108710688