TensorFlow环境配置

1、Anaconda安装

  1. # 清华源下载所需版本.sh安装包,安装
  2. wget https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/
  3. ./xx.sh
  4. # conda常用命令
  5. conda --version
  6. conda info
  7. conda env list
  8. conda list # pip freeze
  9. # 升级当前版本conda
  10. conda update conda
  11. # 创建python==3.6的名称为py36的环境,位置.env/py36
  12. conda create -n py36 python==3.6
  13. # 复制py36环境至new—py36
  14. conda create -n new_py36 --clone py36
  15. # 删除py36环境
  16. conda remove -n new_py36 --all
  17. source activate py36
  18. source deactivate py36
  19. # 根据需要,是否修改环境变量
  20. vim ~/.bashrc
  21. # 最后一行添加
  22. export PATH=/home/anaconda3/bin:$PATH
  23. # 激活环境变量
  24. source ~/.bashrc

2、TensorFlow安装

  • 源码安装

  1. pip install -i https://pypi.douban.com/simple tensorflow==2.3.0

3、Bazel安装

4、Android NDK安装

  1. 解压.zip文件安装(r21

5、adb安装

  1. sudo apt-get install apt

6、安装模型可视化工具netron

  1. sudo apt-get install -f
  2. sudo dpkg -i xxx.deb

7、源码安装cocoapi

  1. $ pip install Cython
  2. $ pip install pycocotools

8、Nvidia驱动安装步骤

Docker是在GPU上运行TensorFlow的最简单方法。

  1. # 安装nvidia-docker
  2. # 检查GPU是否可用
  3. lspci | grep -i nvidia