在linux系统上安装TensorFlow
安装步骤:
1. 在主机host(你的PC)上安装ubuntu1404,并进入该系统。
(Ubuntu 14.04系统下载地址:http://www.linuxidc.com/Linux/2014-04/100352.htm)
2. 在官网或百度云下载jetpack3.1到host上,cd到下载路径,用chmod命令开启该脚本可执行权限,并在终端执行。
$ cd ???
$ chmod 777 jetpack-your-version.run
$ ./jetpack-your-version.run
(官网链接:https://developer.nvidia.com/embedded/jetpack)
(百度云链接:https://pan.baidu.com/s/1Woz6eSHHg4PsGn3hXJXCdg 密码:75vg 失效找我)
3. 打开jetpack后根据提示进行操作,亲测接网线更稳定。
(安装教程参考https://www.jianshu.com/p/bb4587014349)。
4.下载时间和刷机时间大概都要两个小时,完成后转到tx2上操作(用户和密码都是nvidia)。使用tx2要自备蓝牙键盘和鼠标,hdmi显示屏,用vga转hdmi的显示屏时分辨率会出问题,用户体验极差。
5. 在tx2上执行以下命令,下载tensorflow.whl文件。
$ git clone https://github.com/jetsonhacks/installTensorFlowJetsonTX.git
6. 设置系统默认python版本为3.5。
$ sudo update-alternatives --install /usr/bin/python python /usr/bin/python2 100
$ sudo update-alternatives --install /usr/bin/python python /usr/bin/python3 150
$ sudo update-alternatives --config python
#如下
nvidia@tegra-ubuntu:~$ sudo update-alternatives --config pythonThere are 2 choices for the alternative python (providing /usr/bin/python). Selection Path Priority Status------------------------------------------------------------* 0 /usr/bin/python3 150 auto mode 1 /usr/bin/python2 100 manual mode 2 /usr/bin/python3 150 manual modePress to keep the current choice[*], or type selection number: #选0
##修改前
nvidia@tegra-ubuntu:~$ python -VPython 2.7.12
nvidia@tegra-ubuntu:~$ python3 -VPython 3.5.2
##修改后
nvidia@tegra-ubuntu:~$ python -VPython 3.5.2
nvidia@tegra-ubuntu:~$ python3 -VPython 3.5.2
7. 下载pip3,然后用pip3安装tensorflow.whl文件。
$ sudo apt-get install python3-pip
$ pip3 install /home/nvidia/installTensorFlowJetsonTX/TX2/tensorflow-1.3.0-cp35-cp35m-linux_aarch64.whl
8. 检查安装是否成功,没报错就应该是安装成功了。
$ python
>>> import tensorflow
注意事项:
1. jetpack刷机版本
jetpack3.0安装:CUDA8.0 + cuDNNv5.1
jetpack3.1安装:CUDA8.0 + cuDNNv6.0
jetpack3.2安装:CUDA9.0 + cuDNN7.0.5
个人理解:
CUDA8+cuDNN5.x对应安装tensorflow1.2及以下版本
CUDA6/CUDA8+cuDNN6.x对应安装tensorflow1.3-1.4版本
CUDA9+cuDNN7.x对应安装tensorflow1.5-1.6版本
而github上别人编译好的tensorflow.whl文件基本上都是1.3-1.6,因此注意刷机版本。
本人在用jetpack3.0刷机后安装tensorflow1.3,安装成功,但import时报错:
ImportError: libcudnn.so.6: cannot open shared object file: No such file or directo
2. python和pip版本
举例:tensorflow-1.3.0-cp35-cp35m-linux_aarch64.whl ,这个名字里面的cp35就对应python3.5版本,安装前敲指令$ python -V查看一下版本号。最好版本号匹配才安装。
用pip安装时查看$ python -V
用pip3安装时查看$ python3 -V
3. 源码安装
建议不要源码安装,本人在源码安装时,安装bazel经常出错,各种版本不对的问题。而且安装时间更长。
4. 蓝屏问题
先进入字符界面:Ctrl + Alt + F4,进入后会让你进行登录。输入用户名nvidia,密码nvidia
然后安装相应服务,然后重置它
$ sudo apt-get install xserver-xorg-lts-utopic
$ sudo dpkg-reconfigure xserver-xorg-lts-utopic
$ reboot
如果前面第一个操作有问题,需要重置dpkg 后再试,总之按提示操作就好了。
$ sudo dpkg --configure -a
有看到其他人不是安装xserver-xorg-lts-utopic,而是 xserver-xorg-lts-quantal 。估计是版本问题。
出现这种问题的原因之一是暴力关机,请用操作系统关机,不要强制断电。不然有可能导致数据丢失。
如果仍然解决不了,请上网另找教程。
5. apt-get无法使用
E: Sub-process /usr/bin/dpkg returned an error code (1)
$ sudo mv /var/lib/dpkg/info /var/lib/dpkg/info_back
$ sudo mkdir /var/lib/dpkg/info
$ sudo apt-get update
$ sudo apt-get install -f
$ sudo mv /var/lib/dpkg/info /var/lib/dpkg/info_back
$ sudo mv /var/lib/dpkg/info_back /var/lib/dpkg/info