Pytorch 和PyG的安装

最新推荐文章于 2024-02-08 20:01:28 发布

qq_27390023

最新推荐文章于 2024-02-08 20:01:28 发布

阅读量428

点赞数

文章标签： pytorch 深度学习 python

本文链接：https://blog.csdn.net/qq_27390023/article/details/131323976

版权

PyTorch是一个开源的Python机器学习库，基于Torch，用于自然语言处理等应用程序。PyTorch既可以看作加入了GPU支持的numpy，同时也可以看成一个拥有自动求导功能的强大的深度神经网络。PyG（PyTorch Geometric）是一个基于PyTorch的库，用于轻松编写和训练图神经网络（GNN），用于与结构化数据相关的广泛应用。
1. Pytorch 安装

## 下载系统对应版本的miniconda3
curl -O https://repo.anaconda.com/miniconda/Miniconda3-latest-MacOSX-x86_64.sh
sh Miniconda3-latest-MacOSX-x86_64.sh

## bash 安装Miniconda3

## 安装激活conda环境

##  conda 安装pytorch（CPU版本）
conda install pytorch torchvision -c pytorch

## pip 安装安装pytorch（CPU版本）
pip3 install torch torchvision

2. PyG (PyTorch Geometric)安装

# pytorch   version 1.13.1 
conda install pyg -c pyg

# conda 安装后 测试出错
from torch_geometric.nn import GCNConv

# 根据提示
pip uninstall psutil

# 安装成功

错误提示：

ImportError: version conflict: '/Users/zhengxueming/miniconda3/envs/bioinfo/lib/python3.8/site-packages/psutil/_psutil_osx.cpython-38-darwin.so' C extension module was built for another version of psutil (5.9.1 instead of 5.9.5); you may try to 'pip uninstall psutil', manually remove /Users/zhengxueming/miniconda3/envs/bioinfo/lib/python3.8/site-packages/psutil/_psutil_osx.cpython-38-darwin.so or clean the virtual env somehow, then reinstall

参考：

https://pytorch.org/get-started/locally/

http://github.xiaoc.cn/pyg-team/pytorch_geometric

https://zhuanlan.zhihu.com/p/476754252