安装PyTorch-Geometric包之Installation PyG踩坑汇总= =

   由于 项目所需 然后现在不是最快的是  Pyg这个库吗= =,然后我总结了踩坑指南。我要说的是我是想跑这个GNN-for-NLP的代码然后需要安装这个

一、win下的安装

提醒:win下的安装,我当时安装了一半,但是由于博主的电脑有一些限制,我觉得可以按照下面的按照成功,而且我一般用ubun来干活,所以就没继续下去。

官方的安装教程:请戳这里(●'◡'●)

代码跑的是:GNN-for-NLP

安装坑汇总

      (1)(坑1和2)虽然github上写:Compatible with PyTorch 1.x, TensorFlow 1.x and Python 3.x.

      请一定要保证你的环境是   :所有的最新版本!!! 比如cuda=10, PyTorch 1.2及以上(亲测如果不是这个版本,代码跑不了= =,因为我很小白...也可能是因为我太菜了)

      然后查看——版本一定要匹配

$ python -c "import torch; print(torch.version.cuda)"
>>> 10.0

$ nvcc --version
>>> 10.0

(2)将路径加入win下的环境变量(很重要,最大的坑)

          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:...

(3)     No module named 'torch_scatter' 或者 执行不了   pip install --verbose --no-cache-dir torch-scatter (error: Microsoft Visual C++ 14.0 is required. Get it with "Microsoft Visual C++ Build Tools": )

   我当时是因为没有安装  vs2014  ,我的下载链接,你也可以自己找的

  安装成功后,就可以执行这个命令了。请记住 第二步和第三步,一定不能交换,要不然默认安装 cpu版本的 torch-scatter 

这三个坑是 在win下安装我遇到的坑,如果还有请留言,我尽量回复(●'◡'●)。解决了记得回来点赞~~~

二、ubuntu下安装

(1)环境要完全按照代码和官网,此步与win下一致

    当然,如果你服务器cuda是9.0也没关系= = (比如我),里面会报错 sum_cuda" not implemented for 'Bool 

定义到代码get_cc里的最后一行代码,此时需要在 改代码 mask.int().sum() ,凡是有sum的之前都改代码添加int(),然后完美解决 (o´・ェ・`o)

(2)ubuntu下安装没太多坑.....,我遇到的坑都是因为服务器环境太旧(la)了(ji)

      你可以去某宝上租一个服务器,大概2/h把= =,然后跑,然后如果pytorch的版本太旧建议全换,现在pytorch的安装,清华源关闭了,建议执行代码:pip3 install torch torchvision -i https://pypi.mirrors.ustc.edu.cn/simple/

三、总结:

win下安装:

  1.   cuda的环境变量一定要加进去,不如即使按照上也是 cpu版本跑不了
  2.  记得安装vs2014

Linux下安装:

  1. cuda、torch_cuda都必须对应阿阿阿阿阿
  2. 最好新建虚拟环境来安装(●'◡'●)  此处应该有链接  (虚拟环境简直太好用了八)

四、感悟

     在安装这个的时候,我的基础只有一点点,没上过服务器,没跑过pytorch代码,安装这个搞了两天,全会了在服务器跑githhub代码。还有就是 学会了用码云下载github的代码,快和简单极了!!!(●'◡'●),大家具体自己搜一下,怎么用码云下载。

    怕什么真理无穷,进一寸有一寸的欢喜~~

参考

vs2014问题

因为我安装的时候,没有记得一、、记录下来,如果有侵权或者摘抄了某些参考的,可以私信我加上。

 

你可能感兴趣的:(NLP)