一、查看自己的jetpack是什么版本
sudo apt show nvidia-jetpack
二、下载与jetpack版本对应的Tensorflow的.whl文件
点此查看与自己版本对应的tensorflow并下载
三、按NVIDIA官方安装教程安装
大致步骤:
1.设置一个虚拟环境,这里会用到Archiconda,按此教程安装Archiconda,并创建python=3.6的虚拟环境。
2.在虚拟环境中安装依赖(点此官方教程按步骤安装依赖即可,如下图所示)
3.安装Tensorflow
在你下载好的Tensorflow.whl文件所在的路径下
pip3 install tensorflow-2.7.0+nv22.1-cp36-cp36m-linux_aarch64.whl
如果jetpack版本与我的不一致,不要复制上面的,安装你下载的那个版本的Tensorflow.whl。另外如果报pip3 command not found记得:
sudo apt install python3-pip
四、验证一下是否安装成功,打开终端依次输入
python
import tensorflow as tf
tf.__version__
成功输出版本号2.7.0即为安装成功