解决安装包的不兼容问题--torch-scatter

pip install torch-scatter

安装报错:

_CONVERSIONS__ -D__CUDA_NO_HALF2_OPERATORS__ --expt-relaxed-constexpr -O3 --expt-relaxed-constexpr -DTORCH_API_INCLUDE_EXTENSION_H -DTORCH_EXTENSION_NAME=_scatter_cuda -D_GLIBCXX_USE_CXX11_ABI=0 -gencode=arch=compute_86,code=compute_86 -gencode=arch=compute_86,code=sm_86 --use-local-env D:/anaconda3/envs/Hcave/lib/site-packages/torch/include\c10/macros/Macros.h(189): warning C4067: 预处理器指令后有 意外标记 - 应输入换行符 D:\anaconda3\envs\Hcave\lib\site-packages\torch\include\pybind11\detail/common.h(108): warning C4005: “HAVE_SNPRINTF”: 宏重定义 D:\anaconda3\envs\Hcave\include\pyerrors.h(315): note: 参见“HAVE_SNPRINTF”的前一个定义 scatter_cuda.cu nvcc error : 'cudafe++' died with status 0xC00000FD (STACK_OVERFLOW) error: command 'C:\\Program Files\\NVIDIA GPU Computing Toolkit\\CUDA\\v12.5\\bin\\nvcc.exe' failed with exit code 253 [end of output] note: This error originates from a subprocess, and is likely not a problem with pip. ERROR: Failed building wheel for torch-scatter Running setup.py clean for torch-scatter Failed to build torch-scatter ERROR: Could not build wheels for torch-scatter, which is required to install pyproject.toml-based projects.

解决方案：

1. 检查CUDA版本：确保你安装的CUDA版本与PyTorch版本兼容。你提到使用的是CUDA 12.5，但目前PyTorch可能还不支持该版本。可以尝试降级到CUDA 11.7或11.3。

2. 安装预编译的二进制文件：避免从源代码编译，直接安装已经编译好的二进制文件。可以使用如下命令：

   pip install torch-scatter -f https://data.pyg.org/whl/torch-<torch_version>+cu<cuda_version>.html

   例如，如果你使用的是PyTorch 2.0.0和CUDA 11.3：

   pip install torch-scatter -f https://data.pyg.org/whl/torch-2.0.0+cu113.html

3. 增加编译堆栈大小：根据错误信息，cudafe++因为堆栈溢出而失败，可以尝试增加编译器的堆栈大小。在编译前设置环境变量：

   set NVCCFLAGS=-Xcompiler -wd4819,-wd4503,-wd4505 --disable-warnings

4. 更新依赖：确保使用的PyTorch和相关依赖是最新版本，有时库的更新会修复这些问题。可以通过以下命令更新：

   pip install --upgrade torch torchvision torchaudio

5. 虚拟环境：有时候在新的虚拟环境中重新安装库可以解决一些奇怪的问题。可以尝试创建一个新的Anaconda环境，并重新安装PyTorch和torch-scatter：

   conda create -n new_env python=3.6
conda activate new_env
conda install pytorch torchvision torchaudio cudatoolkit=<your_cuda_version> -c pytorch
pip install torch-scatter -f https://data.pyg.org/whl/torch-<torch_version>+cu<cuda_version>.html