TX2安装tensorflow-gpu(亲测有效)

2023-05-16

安装好JetPack 4.3之后,再也不需要从源码对tensorflow进行编译安装,可以通过以下两步安装:
Step1: 安装h5py

sudo apt-get install python3-h5py

主要原因还是我的tx2上pip3安装h5py时会失败.

Step2:安装tensorflow-gpu=1.15.0

sudo pip3 install --pre --extra-index-url https://developer.download.nvidia.com/compute/redist/jp/v43 tensorflow-gpu==1.15.0+nv19.12

Step3:验证安装

import tensorflow as tf
tf.__version__
#1.15.0
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)

TX2安装tensorflow-gpu(亲测有效) 的相关文章

随机推荐