tx2 jetpack4.2 安装tensorflow-gpu

最新推荐文章于 2024-05-21 20:20:40 发布

无提丶

最新推荐文章于 2024-05-21 20:20:40 发布

阅读量1k

点赞数

分类专栏： cuda c++ 文章标签： tx2

本文链接：https://blog.youkuaiyun.com/qq_42608626/article/details/106936783

版权

在TX2上使用JetPack4.2安装TensorFlow-gpu时遇到cudnn版本不兼容的问题，因为4.2自带的cudnn是7.3，但tensorflow1.14及以上需要7.4以上版本。本文档详细介绍了如何安装java、bazel、pip3，并提供了更换cudnn版本的链接，以及从官网下载对应jetpack版本tensorflow-gpu的步骤。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

TX2 (JetPack4.2)安装TensorFlow-gpu有个问题：4.2刷机安装的cudnn是7.3版本的，而tensorflow1.14及其以上需要的是7.4以上的cudnn

安装版本：JetPack4.2
更换cudnn版本：https://blog.youkuaiyun.com/qq_42608626/article/details/106927046

1.安装java

sudo apt-get install openjdk-8-jdk
java -version

配置环境

which javac //查找javac可执行文件
file /usr/bin/javac // 查看链接目标
file /etc/alternatives/javac // 查看链接目标
file /usr/lib/jvm/java-8-openjdk-arm64/bin/javac

sudo gedit ~/.bashrc
// 将下面的代码粘贴到最后
export JAVA_HOME=/usr/lib/jvm/java-8-openjdk-arm64
export JRE_HOME=${JAVA_HOME}/jre
export CLASSPATH=:${JAVA_HOME}/lib:${JRE_HOME}/lib
export PATH=${JAVA_HOME}/bin:$PATH