Jetson Nano环境安装与配置
- 简介
- 环境的安装与配置
- 配置python环境
- 安装pip
- 配置tensorflow-gpu环境
- 确认CUDA已经被正常安装
- 安装所需要的包
- 安装TensorFlow GPU版本
- 测试TensorFlow GPU
简介
NVIDIA® Jetson Nano™ 开发者套件是一款专为创客、学习者和开发者打造的小型 AI 计算机。
环境的安装与配置
配置python环境
Jetson Nano自带python3.6。
安装pip
sudo apt-get install python3-pip python3-dev
安装后pip是9.01版本,需要把它升级到最新版。
python3 -m pip install --upgrade pip #升级pip
运行pip3 -V成功后显示
配置tensorflow-gpu环境
确认CUDA已经被正常安装
nvcc -V
如果能看到CUDA版本号,即为正确安装
安装所需要的包
sudo apt-get install python3-pip libhdf5-serial-dev hdf5-tools
sudo apt-get install python3-scipy
sudo apt-get install python3-pandas
sudo apt-get install python3-sklearn
安装TensorFlow GPU版本
sudo pip3 install -i https://pypi.tuna.tsinghua.edu.cn/simple tensorflow_gpu-1.13.1+nv19.3-cp36-cp36m-linux_aarch64.whl --user
测试TensorFlow GPU
新建一个TestTensorflow.py文件,内容如下:
# -*- coding: utf-8 -*-
import tensorflow as tf
hello=tf.constant('hello,world')
sess=tf.Session()
print(sess.run(hello))