我正在开发一个 PyTorch Geometric 项目,使用 Google Colab 来支持 CUDA。由于默认情况下它的库不存在,因此我运行:
!pip install --upgrade torch-scatter
!pip install --upgrade torch-sparse
!pip install --upgrade torch-cluster
!pip install --upgrade torch-spline-conv
!pip install torch-geometric
最近在导入的时候火炬几何,由于版本升级,出现CUDA版本不匹配的情况:
RuntimeError:检测到 PyTorch 和 torch_sparse 是使用不同的 CUDA 版本编译的。 PyTorch 的 CUDA 版本为 10.1,torch_sparse 的 CUDA 版本为 10.0。请重新安装与您的 PyTorch 安装相匹配的 torch_sparse。
为了解决这个问题,我尝试将 conda 用于特定的 CUDA 版本:
!conda install pytorch==1.4.0 cudatoolkit=10.0 -c pytorch
然而,在跑步时print(torch.version.cuda)
, I get 10.1作为输出而不是我想要的 10.0 。
这是最近的错误,因为它在上周没有抛出这个问题。有解决这个问题的最佳实践吗?
From 他们的网站
Try this
!pip install torch-geometric \
torch-sparse==latest+cu101 \
torch-scatter==latest+cu101 \
torch-cluster==latest+cu101 \
-f https://pytorch-geometric.com/whl/torch-1.4.0.html
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)