由于 项目所需 然后现在不是最快的是 Pyg这个库吗= =,然后我总结了踩坑指南。我要说的是我是想跑这个GNN-for-NLP的代码然后需要安装这个
提醒:win下的安装,我当时安装了一半,但是由于博主的电脑有一些限制,我觉得可以按照下面的按照成功,而且我一般用ubun来干活,所以就没继续下去。
官方的安装教程:请戳这里(●'◡'●)
代码跑的是:GNN-for-NLP
请一定要保证你的环境是 :所有的最新版本!!! 比如cuda=10, PyTorch 1.2及以上(亲测如果不是这个版本,代码跑不了= =,因为我很小白...也可能是因为我太菜了)
然后查看——版本一定要匹配
$ python -c "import torch; print(torch.version.cuda)"
>>> 10.0
$ nvcc --version
>>> 10.0
Add CUDA to $PATH
and $CPATH
(note that your actual CUDA path may vary from /usr/local/cuda
):
具体参考上方官方配置 ,一定要加入环境变量
不然会报错 ——No module named 'torch_scatter.scatter_cuda' ,
因为默认如果你执行
$ pip install --verbose --no-cache-dir torch-scatter
win下没有cuda,它自动帮你安装cpu版本的
$ export PATH=/usr/local/cuda/bin:$PATH //那个位置要看你win下自己的cuda位置,一般应该是在 ../NVI.. GPU 那个文件夹里面
$ echo $PATH
>>> /usr/local/cuda/bin:...
$ export CPATH=/usr/local/cuda/include:$CPATH //那个位置要看你win下自己的cuda位置,一般应该是在 ../NVI.. GPU 那个文件夹里面
$ echo $CPATH
>>> /usr/local/cuda/include:...
我当时是因为没有安装 vs2014 ,我的下载链接,你也可以自己找的
安装成功后,就可以执行这个命令了。请记住 第二步和第三步,一定不能交换,要不然默认安装 cpu版本的 torch-scatter
这三个坑是 在win下安装我遇到的坑,如果还有请留言,我尽量回复(●'◡'●)。解决了记得回来点赞~~~
当然,如果你服务器cuda是9.0也没关系= = (比如我),里面会报错 sum_cuda" not implemented for 'Bool
定义到代码get_cc里的最后一行代码,此时需要在 改代码 mask.int().sum() ,凡是有sum的之前都改代码添加int(),然后完美解决 (o´・ェ・`o)
你可以去某宝上租一个服务器,大概2/h把= =,然后跑,然后如果pytorch的版本太旧建议全换,现在pytorch的安装,清华源关闭了,建议执行代码:pip3 install torch torchvision -i https://pypi.mirrors.ustc.edu.cn/simple/
win下安装:
Linux下安装:
在安装这个的时候,我的基础只有一点点,没上过服务器,没跑过pytorch代码,安装这个搞了两天,全会了在服务器跑githhub代码。还有就是 学会了用码云下载github的代码,快和简单极了!!!(●'◡'●),大家具体自己搜一下,怎么用码云下载。
怕什么真理无穷,进一寸有一寸的欢喜~~
vs2014问题
因为我安装的时候,没有记得一、、记录下来,如果有侵权或者摘抄了某些参考的,可以私信我加上。