安装CPU版本的Pytorch以及依赖库

最近需要用到深度学习的一些算法工具,奈何电脑上并没有条件配置GPU环境,大多数深度学习教程都是基于GPU环境搭建,于是记录一下CPU配置的过程。

首先是基础的torch包的安装,安装之前最好更新一下pip库

/opt/software/python/bin/python3.8 -m pip install --upgrade pip
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu

截至2024.5.10最新的版本是torch 2.3.0(torch · PyPI

另外我还需要安装pytorch_geometric包,注意最新的版本需要基于Python 3.8以上的版本

最新的pytorch_geometric安装地址

Installation — pytorch_geometric documentation

安装代码

pip install torch_geometric
pip install pyg_lib torch_scatter torch_sparse torch_cluster torch_spline_conv -f https://data.pyg.org/whl/torch-2.3.0+cpu.htm
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值