如何安装torch-sparse

如何安装torch-sparse

2020年2月20日，准备用GAT解决一些问题，在网上找到了可以调用GAT的API----torch-geometric，这个包里面实现了GAT，可以直接调用它，然而在安装这个包的时候，有一个依赖的包torch-sparse怎么都安不上，后面在网上找了好久才找到解决办法，特记录一下。

下面只介绍我安装成功的一种方法（我是在服务器上面安装的）

• 首先确保PyTorch是1.4.0的最新版本，如果不是的话请更新到最新版本

  可以执行以下命令查看PyTorch的版本：

   $ python -c "import torch; print(torch.__version__)"
   >>> 1.4.0
  

• 然后确保cuda/bin和cuda/include 在你的 $PATH 和$CPATH 里面，如果不在，请添加进去

  可执行以下命令查看：

  $ echo $PATH
  >>> /usr/local/cuda/bin:...
  
  $ echo $CPATH
  >>> /usr/local/cuda/include:...
  

  例如，我最初执行echo $CPATH的时候就什么也没输出，这时我便需要将cuda/include添加进去，具体方法如下：

  首先回到家目录，然后执行以下命令进入.bashrc里面：

  vim .bashrc
  

  然后将cuda/include添加到$CPATH 里面，在刚才打开的文件最后面添加这一行代码：

  export CPATH="/usr/local/cuda/include:$CPATH"
  

  然后保存并退出，再执行以下命令激活一下即可：

  source .bashrc
  

• 最后便可安装torch-sparse了，命令如下：

  pip install torch-scatter torch-sparse