一、基础依赖
系统:
$ sudo apt-get update
$ sudo apt-get install libhdf5-serial-dev hdf5-tools libhdf5-dev zlib1g-dev zip libjpeg8-dev liblapack-dev libblas-dev gfortran
更新pip3:
$ sudo apt-get install python3-pip
$ sudo pip3 install -U pip testresources setuptools
二、选择特定版本
JP_VERSION :
42 对于JetPack 4.2.2 或 33 对于JetPack 3.3.1
安装 特定版本 的JetPack支持的TensorFlow的最新版本,请发出以下命令:
$ sudo pip3 install --extra-index-url https://developer.download.nvidia.com/compute/redist/jp/v$JP_VERSION tensorflow
TF_VERSION:
TensorFlow的发行版本
NV_VERSION:
NVIDIA容器版本
$ sudo pip3 install --extra-index-url https://developer.download.nvidia.com/compute/redist/jp/v$JP_VERSION tensorflow == $ TF_VERSION + nv $ NV_VERSION
ps:
小技巧 网页查找:https://developer.download.nvidia.cn/compute/redist/jp/$JP_VERSION 所包含的wheel
三、例子
- JetPack 4.2.2
- TensorFlow 1.13.1
- NV 19.03
sudo pip3 install --extra-index-url https://developer.download.nvidia.com/compute/redist/jp/v42 tensorflow-gpu == 1.13.1 + nv19.3
四、其他
官方文档:地址