torch-scatter==1.4.0
torch-sparse==0.4.3
torch-cluster==1.4.5
torch-geometric==1.3.2
在搭DSNet项目的环境时,遇到以上几个torch附属库的安装,分享一下安装时遇到的坑。
首先torch-scatter所依赖的CUDA版本基于你的Pytorch安装版本,可参考此文章。本文亲测成功的一个版本关系为Pytorch(1.1.0)+CUDA(9.0.176)+torch-scatter(1.4.0)+Ubuntu 16.04/windows 10。Pytorch安装命令行为
conda install pytorch==1.1.0 torchvision==0.3.0 cudatoolkit=9.0 -c pytorch
之后输入
pip install torch-scatter==1.4.0
此时有可能出现以下错误:
根据错误信息可知,与VS版本有关,其编译仅支持VS2012、2013、2015、2017版本,由于我安装有更高的2019版本,因此将该版本暂时卸载后即可安装成功。
以此类推,其余几个torch库的版本也都亲测可以成功安装。