一、系统安装排坑
官网自带的最新的系统:
最新的系统有点坑,带的cuda10.2,和其他帖子所说的cuda10.0都不一样,所以你要用旧版本tensorflow的话,就要注意。
https://developer.nvidia.com/embedded/downloads
这个版本是cuda10.0的
注意注意注意: 本工作用的是新的B01版本jetson nano,第一次踩坑就是其他版本的镜像导致不能用,要找对。
二、cuda写入环境变量
sudo vim ~/.bashrc
最后添加:
export CUDA_HOME=/usr/local/cuda-10.0
export LD_LIBRARY_PATH=/usr/local/cuda-10.0/lib64:$LD_LIBRARY_PATH
export PATH=/usr/local/cuda-10.0/bin:$PATH
export CPATH=$CPATH:/usr/local/cuda/include
export LIBRARY_PATH=$LIBRARY_PATH:/usr/local/cuda/lib64
source ~/.bashrc
导入成功
可以用nvcc -V
查看版本,或者用cat /usr/local/cuda/version.txt
来查看
三、必要安装
升级一下apt-get
sudo apt-get update
sudo apt-get full-upgrade
pip安装
sudo apt-get install python3-pip python3-dev
升级到指定版本(坑:这里版本太高会有warning):
python3 -m pip install --upgrade pip==19.0.3
更改pip3文件:
sudo vim /usr/bin/pip3
将原来的:
from pip import main
if __name__ == '__main__':
sys.exit(main())
改为:
from pip import __main__
if __name__ == '__main__':
sys.exit(__main__._main())
查看一下:
pip3 -V
成功: