适用于Python Whl文件的Nvidia Jetson Nano的Yocto Bitbake食谱不在PyPi上

时间:2019-06-12 13:54:56

标签: python python-3.x nvidia yocto bitbake

我正在尝试为NVIDIA特定的PyTorch和Tensorflow Python whl软件包创建2个简单的Yocto Python食谱。目标是Yocto从meta-tegra层为NVIDIA Jetson Nano制作的SD卡映像。没有这些配方,我就可以成功地从meta-tegra编译并引导映像。

NVIDIA本身已经编译并发布了“ .whl” Python软件包,可在以下位置找到它们: https://devtalk.nvidia.com/default/topic/1048776/official-tensorflow-for-jetson-nano-/ https://devtalk.nvidia.com/default/topic/1049071/jetson-nano/pytorch-for-jetson-nano/

我尝试了以下操作,但是两个配方均因各种错误而失败(未找到许可证,缺少setup.py等。)

SUMMARY = "NVIDIA's version of Python Torch"
DESCRIPTION = "NVIDIA's version of Python Torch"
HOMEPAGE = "https://nvidia.com"

SECTION = "devel/python"
LICENSE = "BSD-3-Clause"
LIC_FILES_CHKSUM = "file://LICENSE;md5=79aa8b7bc4f781210d6b5c06d6424cb0"

PR = "r0"
SRCNAME = "Pytorch"

SRC_URI = "https://nvidia.box.com/shared/static/j2dn48btaxosqp0zremqqm8pjelriyvs.whl"

SRC_URI[md5sum] = "9ec85425a64ca266abbfdeddbe92fb18"
SRC_URI[sha256sum] = "3b9b8f944962aaf550460409e9455d6d6b86083510b985306a8012d01d730b8b"

S = "${WORKDIR}/${SRCNAME}-${PV}"

inherit setuptools

CLEANBROKEN = "1"

SUMMARY = "NVIDIA's version of Python Tensorflow"
DESCRIPTION = "NVIDIA's version of Python Tensorflow"
HOMEPAGE = "https://nvidia.com"

SECTION = "devel/python"
LICENSE = "BSD-3-Clause"
LIC_FILES_CHKSUM = "file://generic_BSD-3-Clause;md5=79aa8b7bc4f781210d6b5c06d6424cb0"

PR = "r0"
SRCNAME = "Tensorflow-gpu"

SRC_URI = "https://developer.download.nvidia.com/compute/redist/jp/v42/tensorflow-gpu/tensorflow_gpu-1.13.1+nv19.5-cp36-cp36m-linux_aarch64.whl"

SRC_URI[md5sum] = "ae649a62c274d19d1d096d97284ec2ee"
SRC_URI[sha256sum] = "6639761eccf53cab550d4afb4c8a13dbfe1b1d8051c62e14f83199667ae42d1a"

S = "${WORKDIR}/${SRCNAME}-${PV}"

inherit setuptools

CLEANBROKEN = "1"

我相信我在Yocto中安装了依赖项。如何从这些现有的whl文件创建Yocto食谱?谢谢。

1 个答案:

答案 0 :(得分:1)

可能(未经测试)需要将类似的东西添加到您的食谱中:

DEPENDS += 'pip-native'

do_install() {
    pip install ${S}/tensorflow_gpu-1.13.1+nv19.5-cp36-cp36m-linux_aarch64.whl
}

但可能需要进行更多调整。