win10下python3安装torch_scatter、torch_sparse出错的解决方案

环境说明

软件 版本
os win10
python 3.6
pytorch 1.1.0
cuda 9.2

问题

https://pytorch-geometric.readthedocs.io/en/latest/notes/installation.html
步骤是按照上面链接中来的,但是在执行 $ pip install torch-scatter的时候安装失败。
报错为:

error: command 'C:\Program Files (x86)\Microsoft Visual Studio 14.0\VC\BIN\x86_amd64\cl.exe' failed with exit status 2 failed building wheel for torch scatter

解决方案

查阅了一些资料,可能是CUDA、VS的版本的一些问题。

  1. 找到CUDA\v9.2\include\crt\host_config.h
    一般是c:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.2\include\crt\host_config.h
  2. 找到 #if _MSC_VER < 1600 || _MSC_VER > 1913 3这行代码,修改为#if _MSC_VER < 1600
  3. 再一次进行 pip intall torch-scatter 即可
  4. 安装成功截图
    win10下python3安装torch_scatter、torch_sparse出错的解决方案_第1张图片

参考链接

issues with installation of torch-scatter
Cuda 9.2 does NOT work with Visual Studio 2017 15.7.1

你可能感兴趣的:(环境配置)