昇腾论坛

点赞

收藏

回复

举报

小藤如何安装Tensorflow

发表于2024-05-19 17:53:37

0 查看

正文前感谢昇腾各位工作人员，没有你们的辛勤就没有我们的进步
本文立意交流如何安装Tensorflow
就归纳来说是两个字-要命
真心不想再安装环境上浪费大量的时间精力，因为这样毫无意义
毕竟出去工作，你不能吹嘘说你会再小藤上安装TF，会被嘲笑的
在这里我建议能够提供直接编译好的whl供配置
而不是要自己编写源码来生成whl，再运行，这样非常耗费时间
- 自己用200DK A2编译用了5个小时都还没编完
首先官方固件就支持Pytorch，不支持tf，而且事实上在官方文档也没给出如果自行安装，应该怎么安装
只能按照https://www.hiascend.com/document/detail/zh/CANNCommunityEdition/80RC2alpha002/softwareinst/instg/instg_0043.html

HDF5
- 按照教程去操作就好
h5py依赖包
- Cython
  - 按照教程0.29版本会直接报错
  - 这里本人直接安装最新，并不锁定0.29
- TensorFlow 2.6.5需安装h5py 3.1.0版本
  - 这里直接操作也会报错
  - 这里也不强制安装3.1.0版本，直接安装最新版本
  - 同时要设定要HDF5_DIR环境变量，而非CPATH
  - 可以参照ubuntu安装HDF5 1.10.4 以及对应的python hdf5包_hdf5-1.10.4-CSDN博客

Install Bazel
- 按照只是安装3.7.2
- 提供的文件是bazel-3.7.2-linux-arm64，需要mv到/usr/bin 中并改名bazel
- tensorflow现在已经登不上，用https://tensorflow.google.cn/install/source#install_bazel

安装Clang
- 真希望直接apt-get install能够搞得定，结果大失所望，压根不行
- ```
sudo apt-get update && sudo apt-get install -y llvm-17 clang-17
```
- 直接按照教程，去解压文件，放置到/usr中，最后运行看是否正确安装
- ```
clang --version
```

安装TensorFlow
- 又是一个只能下载源码进行编译的操作
- 从github爬下文件
- 用git checkout 更新到对应的版本 v2.6.5
- 然后修改“nsync-1.22.0.tar.gz”源码包，修改sha值还有包位置
- 然后./configure ,这一步需要注意，一些不要的配置直接就no，不要yes了
- 修改build:op
- 最后用bazel build //tensorflow/tools/pip_package:build_pip_package --repo_env=WHEEL_NAME=tensorflow_cpu
- 进入漫长的等待，写到这里突然想到自己没改sha值..人傻掉，不知道最后会怎么样
- 已经编了5小时不止，要是要重编，我就直接放弃了

本帖最后由匿名用户于 2026/09/03 15:18:45 编辑

匿名回复

我要发帖子