刚在Jetson nano上面配置好了tensorflow,盘点一下我个人遇到的一些坑点,希望能够帮助大家少走弯路
注意事项
1)操作系统镜像刚烧进去开机时选择语言一定要全部选成英文,如果选中文可能会因此出现“build wheel error”之类的问题。
2)尽量使用jetson nano提供的下载源,不要去做没必要的换源。如果某个包真的需要换源才能下载,建议只在下载那个包的适合换源。
3)在下载包前要确定在jetson nano中他的名称,如anaconda在jetson nano适用的版本就和之前的安装方式不同。
4)在下载tensorflow之前不要下载numpy等tensorflow的依赖包,以避免版本冲突带来的问题。
如果您不幸遇到了以上几种问题,能够改正尽量改正,如果改正不了就只好重新烧录了。
安装tensorfow
这里需要注意的是Jetson nano的tensorflow可以按照官网的方式去下载,我的jetson nano自带cuda-10.2和cuda10,第二种方法为官网按照方法
官网如下:
Official TensorFlow for Jetson Nano!
准备工作
1.查看ubuntu版本
cat /etc/issue
或
lsb_release -a
2.检查cuda
cat /usr/local/cuda/version.txt
jetson nano中本身自带cuda,首先要找到cuda所在的文件夹看清楚自己cuda的版本,我这里的版本是cuda10,但是网上的各种配置环境变量都是用的cuda10.0,虽然版本一样,但是环境找不到cuda10,所以配置前一定要看清楚
将符合自己板子的cuda写入环境变量
sudo gedit ~/.bashrc
在打开的文件最后填上
export CUDA_HOME=/usr/local/cuda-10
export LD_LIBRARY_PATH=/usr/local/cuda-10/lib64:$LD_LIBRARY_PATH
export PATH=/usr/local/cuda-10/bin:$PATH
(注:因为博主的板子里面cuda10的文件名称为cuda10而不是cuda10.0,注意配置时写的cuda版本一定要和文件夹名称一模一样)
保存后执行
source ~/.bashrc
3.修改pip
sudo apt-get update