集群服务器安装torch_geometric

  • Post author:
  • Post category:其他




一. 安装torch_geometric步骤

1.直接在终端输入python,就可以打开python, 并且会显示python的版本

python

在这里插入图片描述
2. import torch; print(torch.

version

) 可以显示torch版本cuda版本

import torch

print(torch.__ version__)

在这里插入图片描述

4. 在

pytoch-geometric

选择对应的版本下载, torch_sparse, torch_cluster, torch_scatter,torch_spline_conv

在这里插入图片描述

wget https://data.pyg.org/whl/torch-1.9.0%2Bcu111/torch_cluster-1.5.9-cp38-cp38-linux_x86_64.whl

python -m pip install –user “./torch_cluster-1.5.9-cp38-cp38-linux_x86_64.whl”

5 最后下载torch_geometric

python -m pip install –user torch_geometric



二. 关于集群服务器使用踩坑

1、MobaXterm打开页面和Pychram–Tools–Strat SSH Session打开页面是默认的,不会进入你的设定的python环境,需要导入module来切换

module load anaconda3/py38

2、上面安装torch_geometric只要版本对了就可以了的,出现OSError: libtorch.so: cannot open shared object file: No such file or directory就是因为torch、cuda和torch_cluster等的版本没对上。对于集群服务器就要注意一下,是否你打开的module不对

OSError: libtorch.so: cannot open shared object file: No such file or directory



版权声明:本文为BLvren_原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。