环境说明
| 软件 | 版本 |
|---|---|
| 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的版本的一些问题。
- 找到
CUDA\v9.2\include\crt\host_config.h
一般是c:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.2\include\crt\host_config.h - 找到
#if _MSC_VER < 1600 || _MSC_VER > 1913 3这行代码,修改为#if _MSC_VER < 1600 - 再一次进行
pip intall torch-scatter即可 - 安装成功截图

参考链接
issues with installation of torch-scatter
Cuda 9.2 does NOT work with Visual Studio 2017 15.7.1
在Windows 10环境下使用Python3安装torch_scatter和torch_sparse时遇到错误。按照官方文档步骤操作,但在运行`pip install`时失败,出现报错。解决方案涉及检查CUDA和Visual Studio版本。通过修改环境变量中的CUDA版本号,将`$CUDAVER=9.2`改为`$CUDAVER=10.1`,然后重新运行安装命令,最终成功安装torch_scatter和torch_sparse。
2118

被折叠的 条评论
为什么被折叠?



