第1个回答 2024-08-17
本文为针对Ubuntu系统安装torch_geometric的详细步骤及常见问题解决方案。
一、创建虚拟环境：
为避免系统环境干扰，建议使用虚拟环境进行安装。以CUDA版本12.0为例，使用以下命令创建名为drl_rps的虚拟环境，并激活：
conda create -n drl_rps python=3.6
conda activate drl_rps
二、安装torch：
下载对应版本的torch.whl文件至本地，推荐下载下载地址：download.pytorch.org/wh...。根据Python版本选择文件，如torch-1.7.0+cu110-cp36-cp36m-linux_x86_64.whl，安装过程使用：
pip install torch-1.7.0+cu110-cp36-cp36m-linux_x86_64.whl
安装torchvision同样采用类似命令，文件选择torchvision-0.8.0-cp36-cp36m-linux_x86_64.whl。
三、安装torch_geometric：
访问安装所需辅助包下载地址：pytorch-geometric.com/w...，根据已安装的torch版本和CUDA版本（如torch-1.7.0+cu110），下载对应whl文件（如scatter-sparse-cluster-spline_conv）。
下载完成，使用pip依次安装四个whl文件，确保顺序正确。完成安装后，通过pip install torch_geometric命令安装torch_geometric。
四、检验torch_geometric：
新建example.py文件，从torch_geometric官网复制命令测试。终端执行python example.py，若出现libcusparse.so.11错误提示，需将libcusparse.so.11文件放置于Anaconda安装路径下的lib文件夹，执行以下命令解决：
export LD_LIBRARY_PATH="/home/ubuntu/anaconda3/lib/"
source ~/.bashrc
若在PyCharm中无法调用，将libcusparse.so.11文件复制到虚拟环境lib文件夹，并在PyCharm环境变量中加入相应路径。