torch_geometric安装爬坑记录_20230227

网络查询得知安装torch_geometric需要先安装对应版本的torch-scatter torch-sparse torch-cluster torch-spline-conv
于是我print(torch.__version__)显示版本为1.10.2+cu113
pip 安装对应版本

pip install torch-scatter -f https://pytorch-geometric.com/whl/torch-1.10.2+cu113.html
pip install torch-sparse -f https://pytorch-geometric.com/whl/torch-1.10.2+cu113.html
pip install torch-cluster -f https://pytorch-geometric.com/whl/torch-1.10.2+cu113.html
pip install torch-spline-conv -f https://pytorch-geometric.com/whl/torch-1.10.2+cu113.html

之后pip install torch_geometric

结果！！！
import toprch_sparse时
linux下报错OSError: libcusparse.so.11: cannot open shared object file: No such file or directory
我又按照网上提示修改path
由于我是在实验室的服务器上 服务器有多个用户
尝试执行nvcc -V 提示没有命令 于是我vi ~/.bashrc在打开文件的最下面添加了我的服务器的cuda路径/usr/local/cuda-10.2

export PATH="/usr/local/cuda-10.2/bin:$PATH"
export LD_LIBRARY_PATH="/usr/local/cuda-10.2/lib64:$LD_LIBRARY_PATH"

source激活生效

source ~/.bashrc

现在nvcc -V可以正常显示了 但是报错还在
我发现在我的虚拟环境pytorch-gpu的lib里已经有libcusparse.so.11了
于是我vi ~/.bashrc在打开文件的最下面添加

export LD_LIBRARY_PATH="/storage/tom/.conda/envs/pytorch-gpu/lib:$LD_LIBRARY_PATH"

source激活生效

source ~/.bashrc

import toprch_sparse没有报错了