【Jetson-Nano】2.Tensorflow和Pytorch的安装

文章目录

• 1、Tensorflow多版本安装
• • 1.1 Protobuf 安装
  • 1.2 安装依赖包及tensorflow1.15
  • 1.3 安装其它常用库
  • 1.4 测试python包是否安装成功
  • 1.5 TensorRT和Opencv的安装
  • 1.6 pycuda和onnx安装
  • 1.7 Tensorflow2.3安装
• 2、Pytorch安装
• • 2.1 安装pytroch和torchvision
  • 2.2 安装环境验证
• 参考内容

首先按照我的其它 博客完成jetson nano 基础环境的配置。然后就可以对深度学习框架进行安装。要注意的是，nano每次断电再开机后，系统时间是错误的，要先手动进行设置。

1、Tensorflow多版本安装

创建虚拟环境并激活：

mkvirtualenv nano
workon nano

• 1
• 2

Tensorflow不同的版本将安装在不同的虚拟环境中。

1.1 Protobuf 安装

退出虚拟环境，在系统中安装Protobuf。
首先配置protobuf,这样可以使tensorflow运行更快。如果protobuf和libprotobuf没有安装，Tensorflow的性能会被大打折扣。当然，使用pip来安装tensorflow 的时候，会自动安装一个protobuf的版本，但是可能不是很合适。关于这部分性能的影响参见