from torch.utils.cpp_extension import BuildExtension, CUDAExtension ModuleNotFoundError: No module named 'torch'
时间: 2023-12-03 17:40:50 浏览: 411
根据提供的引用内容,可以看出您遇到了两个问题。第一个问题是安装了torch_scatter模块后仍然出现“ModuleNotFoundError: No module named ‘torch_scatter’”的错误。第二个问题是在导入torch.utils.cpp_extension模块时出现“ModuleNotFoundError: No module named 'torch'”的错误。
针对第一个问题,可能是因为您的torch_scatter模块安装不正确或者版本不兼容。您可以尝试使用以下命令重新安装torch_scatter模块:
```shell
pip uninstall torch_scatter
pip install torch_scatter -f https://pytorch-geometric.com/whl/torch-1.9.0+cu102.html
```
其中,-f选项指定了安装的版本,这里选择了适用于torch 1.9.0和CUDA 10.2的版本。您可以根据自己的环境选择合适的版本。
针对第二个问题,可能是因为您的torch模块没有正确安装或者没有添加到系统路径中。您可以尝试使用以下命令重新安装torch模块:
```shell
pip uninstall torch
pip install torch
```
如果您已经安装了torch模块,可以尝试将torch模块所在的路径添加到系统路径中。例如,在Linux系统中,可以使用以下命令将torch模块所在的路径添加到系统路径中:
```shell
export PYTHONPATH=/path/to/torch:$PYTHONPATH
```
阅读全文