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

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

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

环境说明

软件版本
oswin10
python3.6
pytorch1.1.0
cuda9.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. 安装成功截图
    安装成功截图

参考链接

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

评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值