简介
我试图在Google Colab上安装PyTorch Geometric，但这并不简单。即使检查了各种内容，也没有编写安装方法，所以我将其保留为备忘。
什么是Google合作实验室？
Google Colaboratory(Google Colab和Colaboratory两者)是Jupyter Notebook环境，可以在Google提供的云上执行??。最重要的是，您可以免费使用Tesla K80 GPU。对于不属于研究机构或公司的个人来说，这是使用GPU的最佳环境。
[如何以每秒的速度使用免费GPU]在Colaboratory上的深度学习实践提示中详细介绍了如何使用它。
什么是PyTorch Geometric？
PyTorch Geometric(PyG)是基于PyTorch的图形神经网络库。提供了诸如GCN和GAT之类的图神经网络的实现，以及诸如Node2vec之类的图嵌入方法。
为了使用PyTorch Geometric，除了torch-geometric之外，还需要安装四个库(torch-scatter，torch-sparse，torch-cluster，torch-spline-conv)。但是，安装这四个库成为一个问题。
无法安装...
无法安装除PyTorch Geometric之外的其他库，例如PyTorch散布图。例如，当我尝试安装Torch-scatter时，出现如下错误消息：
1 2 3 4 5 6 7 8 9 10 | Collecting torch-scatter Downloading https://files.pythonhosted.org/packages/30/d9/1d5fd4d183dabd9e0a1f7008ecf83318432359f4cc27480e3f2212f44d9c/torch_scatter-1.3.2.tar.gz Building wheels for collected packages: torch-scatter Building wheel for torch-scatter (setup.py) ... error ERROR: Failed building wheel for torch-scatter Running setup.py clean for torch-scatter Failed to build torch-scatter Installing collected packages: torch-scatter Running setup.py install for torch-scatter ... error ERROR: Command errored out with exit status 1: /usr/bin/python3 -u -c 'import sys, setuptools, tokenize; sys.argv[0] = '"'"'/tmp/pip-install-ttwepv79/torch-scatter/setup.py'"'"'; __file__='"'"'/tmp/pip-install-ttwepv79/torch-scatter/setup.py'"'"';f=getattr(tokenize, '"'"'open'"'"', open)(__file__);code=f.read().replace('"'"'\r\n'"'"', '"'"'\n'"'"');f.close();exec(compile(code, __file__, '"'"'exec'"'"'))' install --record /tmp/pip-record-e9z1yo5h/install-record.txt --single-version-externally-managed --compile Check the logs for full command output. |
为什么？
显然原因是
- 硬件加速器未设置为GPU
- gcc，g的版本不正确
- CPATH不通过
分3分。
解决方案
1。编辑>笔记本设置>使用硬件加速器更改为GPU
2。将gcc和g的版本减少到5。通过CPATH。
1 2 3 4 5 6 | !apt install gcc-5 g++-5 -y !ln -sf /usr/bin/gcc-5 /usr/bin/gcc !ln -sf /usr/bin/g++-5 /usr/bin/g++ !export CPATH=/usr/local/cuda/include:$CPATH |
3。安装Pytorch Geometric和随附的库
1 2 3 4 5 | !pip install torch-scatter !pip install torch-sparse !pip install torch-cluster !pip install torch-spline-conv !pip install torch-geometric |
终于
PyTorch Geometric现在可以在Colab上使用，但是...安装时间很长...
就这样！
参考链接
- 火炬散布
- CPATH为空#64