报错:No module named ‘torch_scatter.scatter_cuda‘ 适用gpu版的pytorch环境的torch_geometric安装

2020/9/16 安装 
安装目的:跑CGC-Net(FATURE_EXTRACT)
报错:No module named 'torch_scatter.scatter_cuda' 

参考:https://www.cnblogs.com/yqpy/p/13498200.html
参考:https://zhuanlan.zhihu.com/p/138826540
参考:https://blog.csdn.net/m0_37847767/article/details/103160616

报错原因:GCG-Net里面的torch-scatter、torch-sparse、torch-spline-conv、torch-cluster(requirements.txt版本)
都是cpu版本,我的是cuda,就是gpu版本,故它找不到scatter_cuda

解决方法:参考https://pytorch-geometric.com/whl/torch-1.4.0.html里这4个包,只有cuda10.1是符合win操作系统的python3.6版本
下载对应的cuda和cudnn

安装cuda 10.1
安装cudnn v8.0.?(10.1版本)
在环境里 (环境activate prt_1.4.0_cu10.1_vs_0.5.0)
conda install pytorch==1.4.0 torchvision==0.5.0 cudatoolkit=10.1 -c pytorch
conda list,发现conda环境里安装的cudnn是7.0?

注意:适用gpu版的pytorch环境的torch_geometric安装
从https://pytorch-geometric.com/whl/torch-1.4.0.html下载
torch_cluster-1.5.2+cu101-cp36-cp36m-win_amd64.whl
torch_scatter-latest+cu101-cp36-cp36m-win_amd64.whl
torch_sparse-latest+cu101-cp36-cp36m-win_amd64.whl
torch_spline_conv-latest+cu101-cp36-cp36m-win_amd64.whl
放到C:\Users\Diting\anaconda3\envs\prt_1.4.0_cu10.1_vs_0.5.0\Lib\site-packages目录下
在该目录cmd,然后active环境:activate prt_1.4.0_cu10.1_vs_0.5.0
pip install 上述4个包
最后pip install torch_geometric==1.4.2 (requirements.txt里的版本)
最后把gcg所需要的其他安装包安装上
pip install -i https://pypi.doubanio.com/simple/ -r requirements.txt

打工熬成

你可能感兴趣的:(pytorch,深度学习)