pytorch-aarch64

Build Status Coverage

PyTorch torchvision torchaudio torchtexttorchcsprng 的安装包 (whl) ,适用于 aarch64 / ARMv8 / ARM64 设备

English (for GitHub) Gitee Web

安装

conda 🆕 (测试中)

conda install -c kumatea numpy pytorch

现已可用:

pip

pip install torch -f https://cf.torch.kmtea.eu/whl/stable-cn.html

备用链接:pip install torch -f https://torch.kmtea.eu/whl/stable-cn.html

如有需要,添加 torchvision torchaudio torchtext torchcsprng 等包。

注意:此命令安装的是最新版本。 要选择特定的版本,请查看 对应版本 部分。

如果需要手动选择whl文件,请查看 release

Docker (即将弃用)

docker run -it kumatea/pytorch

运行 docker pull kumatea/pytorch 以拉取镜像。

点击此处 查看所有版本。

FastAI

FastAI是一个基于PyTorch的优秀开源高级深度学习框架。

conda (推荐)

conda install -c fastai -c kumatea fastai

同样,fastbook 也可以用以下方法安装:

conda install -c fastai -c kumatea fastbook

pip

pip install fastai -f https://cf.torch.kmtea.eu/whl/stable.html

torchtorchvision 将作为依赖自动安装。


附加信息

CUDA / CUDNN 支持

由于编译环境(见下)不包含 Nvidia GPU,无法支持 cuda 。

如有需要,请使用 Nvidia Jetson 开发板运行 编译代码

编译环境

SoC:BCM2711 (四核A53)

架构:ARMv8 / ARM64 / aarch64 架构

操作系统:Debian Buster

虚拟化:Docker