0
点赞
收藏
分享

微信扫一扫

Jetson Nano环境安装与配置Tensorflow-gpu1.13.1


Jetson Nano环境安装与配置

  • ​​简介​​
  • ​​环境的安装与配置​​
  • ​​配置python环境​​
  • ​​安装pip​​
  • ​​配置tensorflow-gpu环境​​
  • ​​确认CUDA已经被正常安装​​
  • ​​安装所需要的包​​
  • ​​安装TensorFlow GPU版本​​
  • ​​测试TensorFlow GPU​​

简介

NVIDIA® Jetson Nano™ 开发者套件是一款专为创客、学习者和开发者打造的小型 AI 计算机。

Jetson Nano环境安装与配置Tensorflow-gpu1.13.1_cuda


Jetson Nano环境安装与配置Tensorflow-gpu1.13.1_Jetson Nano_02


Jetson Nano环境安装与配置Tensorflow-gpu1.13.1_tensorflow_03


Jetson Nano环境安装与配置Tensorflow-gpu1.13.1_python_04

环境的安装与配置

配置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成功后显示

Jetson Nano环境安装与配置Tensorflow-gpu1.13.1_cuda_05

配置tensorflow-gpu环境

确认CUDA已经被正常安装

nvcc -V

如果能看到CUDA版本号,即为正确安装

Jetson Nano环境安装与配置Tensorflow-gpu1.13.1_cuda_06

安装所需要的包

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))

Jetson Nano环境安装与配置Tensorflow-gpu1.13.1_Jetson Nano_07


Jetson Nano环境安装与配置Tensorflow-gpu1.13.1_python_08


举报

相关推荐

0 条评论