在conda环境中安装torch-geometric

torch-geometric是一个基于PyTorch的图神经网络库，可用于处理图形数据并训练GNN模型。在新建的conda环境中，我们将遵循以下步骤来安装torch-geometric：
步骤1：新建conda环境（可选）
如果尚未创建conda环境，你可以使用以下命令创建一个名为“geo1”的新环境：

conda create -n geo1 python=3.6

然后激活新创建的环境：

conda activate geo1

步骤2：安装pytorch和对应驱动
首先，确保你的系统已经安装了正确版本的NVCC（NVIDIA CUDA Compiler），这里我们选择版本为10.0。你可以使用以下命令检查NVCC的版本：

nvcc --version

接下来，根据你的系统配置，选择适合的PyTorch版本进行安装。在本例中，我们将安装PyTorch 1.4.0版本，并使用cu100驱动。运行以下命令进行安装：

pip install torch==1.4.0+cu100 torchvision==0.5.0+cu100 -f https://download.pytorch.org/whl/torch_stable.html

步骤3：安装torch-geometric库
一旦PyTorch和驱动程序安装完成，就可以开始安装torch-geometric库了。运行以下命令进行安装：

pip install torch-geometric

步骤4：测试代码
最后，为了验证安装是否成功，你可以运行以下Python代码来测试torch-geometric库是否可以正常工作：

import torch
from torch_geometric.nn import GCNConv, Sequential

以上代码会导入PyTorch和torch-geometric中的一些基础模块，然后通过打印输出模块的名称来验证安装是否成功。如果看到“”和“”等输出信息，则说明安装成功。至此，你已经成功在conda环境中安装了torch-geometric库，并可以使用它来构建和训练图神经网络模型了。