时隔四天,tensorflow是终于安装好了,刷机完之后天真的以为tensorflow很好安装,结果被TX2狠狠的打了脸,总之还是自己学艺不精。
刷完机,老师要求在TX2上安装python3.7版本,装好之后也把python3指向的默认版本改成了python3.7,接下来就是安装tensorflow,搞了四天,真的崩溃了,第一次安装tensorflow以为随便找个版本就好,结果就下载了一个找到的最高的版本tensorflow1.13,结果运行pip安装之后,错误一大堆,查了很多资料后才发现jetpack3.2刷机的TX2只能安装1.5-1.8,并且TX2所需要的tensorflow是aarch,和一般linux安装的大不相同,好的,这些东西理清楚后,又发现网上的版本只有python2.7和python3.5的tensotflow,而我的默认python已经设置为python3.7,无奈又得重新改回python3.5,但是改着改着,就发现许多依赖包没了,pip的python2版本又下载不了,于是,,,又重新刷机了,这次改为jetpack3.3刷机。
一、重新刷机(jetpack3.3)
刷机步骤参考以下博客:
https://blog.csdn.net/nulidehahafr/article/details/97754593
二、安装tensorflow
看了网上的安装方式,大概有两种:直接安装和编译安装。我第一次用的编译安装,但是需要内存很大,要分一个8G的虚拟内存,我运行到最后一步出了一个错误,但一直没找到解决方法,就换成直接安装。因此就只说一下直接安装。
1、安装cuda9.0
昨天看博客有博主说刷机后在TX2上并没有安装好cuda和cudnn,所以一般得再次安装一遍。我们刷机时,主机上有个jetpack_download文件夹,里面就有cuda和cudnn的安装文件,因此,我们把这个文件夹直接拷贝到TX2上,然后运行以下命令:
//打开新终端
sudo apt-key add /var/cuda-repo-9-0-local/7fa2af90.pub
//cd进入jetpack_download文件夹
sudo dpkg -i cuda-repo-l4t-9-0-local_9.0.252-1_arm64.deb
sudo dpkg -i libcudnn7_7.1.5.14-1+cuda9.0_arm64.deb
sudo dpkg -i libcudnn7-dev_7.1.5.14-1+cuda9.0_arm64.deb
sudo apt update
sudo apt install cuda-toolkit-9.0
运行nvcc -V,若输出cuda的相关信息,则安装成功,如果提示没有nvcc指令,那就从头再来吧。。。
2、安装pip3
TX2刷机之后自带python2.7和python3.5,我安装的tensorflow是基于python3.5的,因此需要安装pip3.
https://bootstrap.pypa.io/在该链接中下载get-pip.py,运行
sudo python3 get-pip.py
再用pip -V测试pip是否安装好
3、安装tensorflow
进入NVIDIA官网(如下链接)下载tensorflow_gpu-1.9.0
https://developer.nvidia.com/embedded/downloads#?search=tensorflow&tx=$product,jetson_tx2
//安装系统包
sudo apt-get install libhdf5-serial-dev hdf5-tools libhdf5-dev zlib1g-dev zip libjpeg8-dev
//安装python包
sudo pip3 install -U numpy grpcio absl-py py-cpuinfo psutil portpicker six mock requests gast h5py astor termcolor protobuf keras-applications keras-preprocessing wrapt google-pasta
sudo apt-get install python-dev python3-dev
//安装tensorflow
sudo pip3 install tensorflow_gpu-1.9.0+nv18.8-cp35-cp35m-linux_aarch64.whl
在安装tnesorflow的时候,在ubuntu里下载的是一个压缩包,就直接用压缩包安装,千万别解压之后再安装,因为解压之后文件后缀就会变成.whl_FILE,安装这个会出现很多错误。(今天查了一下tensorflow官网,Google使用wheel标准分发tensorflow,它就是.whl后缀的ZIP格式文件,并且tensorflow推荐使用pip install)我就是这样安装了好几天,错误在网上也找不到解决方法。并且在这一步网络状态一定要好,昨天早上装了一早上都是网络错误,吃了饭过来再试了一次就一下安装好了。
4、验证tensorflow
打开一个终端,运行python3,然后导入tensorflow,不报错就表示安装成功
import tensorflow as tf
tf.__version__
输出:1.9.0
5、卸载tensorflow(这步是你需要卸载它的时候再运行)
pip3 uninstall -y tensorflow-gpu
6、升级tensorflow
pip3 install--upgrade tensorflow