安装torch-geometric

原创已于 2023-02-04 15:34:48 修改 · 1.4k 阅读

5 ·

CC 4.0 BY-SA版权

文章标签：

#pytorch #深度学习 #python

于 2023-02-03 14:58:31 首次发布

安装配置专栏收录该内容

9 篇文章

订阅专栏

本文提供了两种安装PyTorch的方法，包括通过官网命令和本地whl文件。接着详细阐述了如何安装与Torch兼容的四个相关库torch_cluster、torch_scatter、torch_sparse和torch_spline_conv，强调需注意Python和CUDA版本。最后，介绍了如何安装torch-geometric库。

一、安装torch

方式一：在pytorch官网查找命令
方式二：下载对应版本的whl到本地，进入whl所在文件夹使用pip安装
※ 注意torch1.8及以后才可以使用torch_geometric
e.g.:

pip install torch-版本信息.whl

二、安装四个相关库

从pytorch-geometric的whl网站中找到对应torch和cuda版本的4个whl，下载完毕后手动pip安装。
注意看好环境的python版本信息，cp3x对应python3.x

e.g.:（自己安装时记得替换相关版本号）

pip install torch_cluster-1.5.9-cp38-cp38-win_amd64.whl
pip install torch_scatter-2.0.9-cp38-cp38-win_amd64.whl
pip install torch_sparse-0.6.12-cp38-cp38-win_amd64.whl
pip install torch_spline_conv-1.2.1-cp38-cp38-win_amd64.whl