见:ModuleNotFoundError: No module named ‘torch_sparse.unique_cuda’
原因:没有安装 CUDA,CUDANN, 或者没有配好环境变量。
虽然用
conda install cudatoolkit
可以让 pytorch 使用 gpu 版本,但是系统环境里是没有安装 CUDA 的链接库的,导致使用 pip 安装 torch_sparse 的时候,源码编译时没有找到系统里的 CUDA 库的路径,就默认安装 cpu 版本的torch_sparse,而省略 cuda 相关的模块。

本文探讨了在安装torch_sparse库时遇到的ModuleNotFoundError问题,深入分析了未正确配置CUDA环境的原因,以及如何通过调整环境变量和确保系统中存在CUDA链接库来解决此问题。
1万+

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



