在colab中为torch\u sparse构建轮子需要永远的时间

wdebmtf2  于 2021-07-13  发布在  Java
关注(0)|答案(1)|浏览(484)

我尝试在google colab上运行以下内容: !pip install torch_sparse 一开始,它似乎运行良好:

Collecting torch_sparse
  Downloading https://files.pythonhosted.org/packages/9a/86/699eb78ea7ce259da7f9953858ec2a064e4e5f5e6bf7de53aa6c8bb8b9a8/torch_sparse-0.6.9.tar.gz
Requirement already satisfied: scipy in /usr/local/lib/python3.7/dist-packages (from torch_sparse) (1.4.1)
Requirement already satisfied: numpy>=1.13.3 in /usr/local/lib/python3.7/dist-packages (from scipy->torch_sparse) (1.19.5)
Building wheels for collected packages: torch-sparse

但从这里开始,它就永远保持运行,不输出任何错误信息,轮子永远也造不出来。
我的colab环境是托管的,并使用gpu加速器。我也可以安装 Torch 和初始化cuda事先,但它不会改变任何事情。

wqnecbli

wqnecbli1#

安装实际上是在30分钟到1小时后完成的(我没有确切的时间)。然而,当尝试导入torch时,我在这里描述了一个问题:googlecolab上的pytorch几何cuda安装问题
我尝试应用最流行的答案,但由于它似乎已过时,我将其更新为以下内容:

!pip install torch-geometric \
  torch-sparse \
  torch-scatter \
  torch-cluster \
  -f https://pytorch-geometric.com/whl/torch-1.8.0+cu101.html

这个安装工作得很好,只花了几秒钟。到目前为止,它似乎使我能够从torch\u geometric/torch\u sparse导入所需的一切。如果我以后遇到任何问题,我会告诉你的!

相关问题