集群服务器安装torch_geometric

文章目录

  • 一. 安装torch_geometric步骤
  • 二. 关于集群服务器使用踩坑

一. 安装torch_geometric步骤

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

python

在这里插入图片描述2. import torch; print(torch.version) 可以显示torch版本cuda版本

import torch
print(torch.__ version__)

集群服务器安装torch_geometric_第1张图片
4. 在pytoch-geometric选择对应的版本下载, torch_sparse, torch_cluster, torch_scatter,torch_spline_conv
集群服务器安装torch_geometric_第2张图片

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

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