windows10下torch-geometric安装踩坑

近期安装torch-geometric的时候踩了一些坑,在这里简单梳理一下安装过程

Linux下可以直接pip install torch-geometric, 但在windows下会报错提示需要C++编译器, 我们可以通过安装对应版本的VS来获取.
我使用的pytorch版本为1.10,CUDA 9.0, 在CUDA9.0\development\include\crt\host_config.h中可以看到它需要版本不为 _MSC_VER < 1600 || _MSC_VER > 1911才可以正确编译. 我们可以通过命令行运行cl.exe查看编译器版本

这里的19.00.24215.1表示我的_MSC_VER为1900. 通过安装VS2015社区版 (记得安装时勾选C++) 可以获得这一版本的编译器. 如果电脑中安装的是VS2017或者更新的VS, 版本可能为1916或者更大

需要注意的是, 网上一些其他文章有通过暴力修改host_config.h中的值1911为你的编译器版本来跳过这一判断. 我使用这个方法时虽然不会提示版本不对, 但后续仍会出现CUDA编译出错的问题.

建议先完全卸载掉其他版本的VS, 然后安装对应版本. 注意, 直接使用VS安装器的卸载功能可能无法完全卸载掉部分模块, 导致重新安装时提示部分功能未成功安装, 我们可以在控制面板的卸载程序中找到所有Microsoft Visual C++ 20** Redistributable开头的程序并卸载即可.

正确安装好VS后即可成功安装torch-geometric