Google Colab上的PyTorch Geometric CUDA安装问题

Question

我正在使用Google Colab进行CUDA支持的PyTorch Geometric项目。由于默认情况下不存在该库，因此我运行:

!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

最近，由于版本升级，在导入 torch_geometric 时，存在CUDA版本不匹配的说法:

RuntimeError: Detected that PyTorch and torch_sparse were compiled with different CUDA versions. PyTorch has CUDA version 10.1 and torch_sparse has CUDA version 10.0. Please reinstall the torch_sparse that matches your PyTorch install.

为了解决这个问题，我尝试将conda用于特定的CUDA版本，如下所示:
!conda install pytorch==1.4.0 cudatoolkit=10.0 -c pytorch
但是，在运行 print(torch.version.cuda)时，我得到 10.1 作为输出，而不是我想要的10.0。

这是最近的错误，因为它在上周没有引发此问题。有解决此问题的最佳做法吗？

Answer 1

从their website

试试这个

!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