问题:
安装pytorch-geometric报错,有两个报错问题:
-
torch.cluster和torch.scatter无法使用cuda
-
torch.cluster版本问题,提示找不到版本软链接
解决方法
不能从anaconda安装pytorch,用官方稳定版本的安装包,pip安装即可
一致性版本包的官方解答:
this problem will be solved. All dependent packages are listed:
cu101/torch-1.4.0-cp38-cp38-linux_x86_64.whl
cu101/torchvision-0.5.0-cp38-cp38-linux_x86_64.whl
torch_cluster-1.5.3+cu101-cp38-cp38-linux_x86_64.whl
torch_scatter-2.0.4+cu101-cp38-cp38-linux_x86_64.whl
torch_sparse-0.6.1+cu101-cp38-cp38-linux_x86_64.whl
torch_spline_conv-1.2.0+cu101-cp38-cp38-linux_x86_64.whl
本文解决PyTorch-Geometric安装过程中遇到的CUDA兼容性问题,包括torch.cluster和torch.scatter无法使用CUDA及版本软链接问题。提供了一致性版本包的官方解答,推荐使用官方稳定版本并通过pip安装。
1万+

被折叠的 条评论
为什么被折叠?



