安装conda+python+pytorch+pyg+ogb指定版本(最全面)

作者:Nicky2024.01.17 23:18浏览量:40

简介:本文将详细介绍如何使用conda在Windows系统上安装指定版本的Python、PyTorch、PyTorch Geometric和Open Graph Benchmark,以便进行深度学习和其他机器学习任务。

首先,我们需要安装Anaconda,这是一个流行的Python发行版,包含了conda包管理器和其他科学计算所需的库。可以从Anaconda官网下载并安装。
安装完成后,打开Anaconda Prompt,然后创建一个新的conda环境。假设我们要安装的Python版本是3.8.8,环境名称为neocpu,可以运行以下命令:
conda create -n neocpu python=3.8.8
如果遇到镜像源问题,导致404错误,可以尝试删除conda的镜像源。运行以下命令:
conda config —remove-key channels
接下来,激活新创建的环境。在Windows上,运行以下命令:
activate neocpu
现在,我们可以在这个环境中安装PyTorch。首先,我们需要确定要安装的PyTorch版本。假设我们要安装1.8.1版本。然后,运行以下命令:
conda install pytorch=1.8.1
接着,检查电脑是否带有GPU。如果带有GPU,还需要查看CUDA版本。运行以下命令查看CUDA版本:
nvcc -V
这里假设我们直接使用CPU进行计算。接下来,安装PyTorch Geometric(PyG),这是一个专门为图神经网络设计的PyTorch库。由于PyG需要与PyTorch版本相对应,因此我们需要找到与PyTorch 1.8.1版本相对应的PyG版本。假设我们要安装PyG 2.0.3版本,可以运行以下命令:
conda install pytorch-geometric=2.0.3
最后,安装Open Graph Benchmark(OGB),这是一个用于评估图神经网络性能的基准测试。我们可以使用pip来安装OGB。运行以下命令:
pip install ogb
现在,我们已经成功安装了conda、Python、PyTorch、PyTorch Geometric和Open Graph Benchmark。在neocpu环境中,我们可以使用这些库来进行深度学习和其他机器学习任务。请注意,以上步骤可能会因操作系统和网络环境的不同而略有差异。