< >
Home » Xavier入门教程软件篇 » Xavier入门教程软件篇-安装TensorFlow-GPU

Xavier入门教程软件篇-安装TensorFlow-GPU

Xavier入门教程软件篇-安装TensorFlow-GPU

说明:

  • 介绍如何为xavier安装TensorFlow-GPU

步骤:

  • 安装依赖包:
$ sudo apt-get install libhdf5-serial-dev hdf5-tools
$ sudo apt-get install python3-pip
$ pip3 install -U pip
$ sudo apt-get install zlib1g-dev zip libjpeg8-dev libhdf5-dev 
$ sudo pip3 install -U numpy grpcio absl-py py-cpuinfo psutil portpicker grpcio six mock requests gast h5py astor termcolor
  • 安装默认最新的包
$ pip3 install --pre --extra-index-url https://developer.download.nvidia.com/compute/redist/jp/v42 tensorflow-gpu
  • 或安装不同的版本
$ pip3 install --extra-index-url https://developer.download.nvidia.com/compute/redist/jp/v42 tensorflow-gpu==$TF_VERSION+nv$NV_VERSION
  • TF_VERSION:The released version of TensorFlow, for example, 1.12.0.
  • NV_VERSION:The monthly NVIDIA container version of TensorFlow, for example, 19.01.
  • 例如,安装 TensorFlow1.12.0,容器版本 19.01:
$ pip3 install --extra-index-url https://developer.download.nvidia.com/compute/redist/jp/v42 tensorflow-gpu==1.12.0+nv19.1

测试:

  • 启动python
$ python3
  • 导入
>>> import tensorflow

参考:

  • https://devtalk.nvidia.com/default/topic/1042125/official-tensorflow-for-jetson-agx-xavier/
  • https://docs.nvidia.com/deeplearning/dgx/install-tf-xavier/index.html

纠错,疑问,交流: 请进入讨论区点击加入Q群

获取最新文章: 扫一扫右上角的二维码加入“创客智造”公众号


标签: xavier入门教程软件篇