安装torch-scatter的坑

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

 此时有可能出现以下错误:

安装torch-scatter的坑_第1张图片

 根据错误信息可知,与VS版本有关,其编译仅支持VS2012、2013、2015、2017版本,由于我安装有更高的2019版本,因此将该版本暂时卸载后即可安装成功。

安装torch-scatter的坑_第2张图片

 以此类推,其余几个torch库的版本也都亲测可以成功安装。

你可能感兴趣的:(python,pytorch)