?????????????????
nvidia jetson agx xavier这个小家伙目前释放出最新的jetpack 4.2.1,也是我的刷机版本,刷机之后自带cuda10,所以我的xavier环境:
Ubuntu 18.04 LTS aarch64
jetpack 4.2.1
python 3.6
cuda 10
官方给出的组分:
see?一样滴。
搭建tf-gpu,直接:
sudo pip3 install --pre --extra-index-url https://developer.download.nvidia.com/compute/redist/jp/v42 tensorflow-gpu --user
相关依赖包会被自动安装。但如果在依赖包安装过程中h5py报错(即我遇到的问题),则:
sudo apt-get install libhdf5-serial-dev hdf5-tools
或者可以先装载依赖包:
sudo apt-get install libhdf5-serial-dev hdf5-tools
sudo apt-get install python3-pip
pip3 install -U pip
sudo apt-get install zlib1g-dev zip libjpeg8-dev libhdf5-dev
sudo pip3 install -U numpy grpcio absl-py py-cpuinfo psutil portpicker grpcio six mock requests gast h5py astor termcolor
然后再执行安装:
sudo pip3 install --pre --extra-index-url https://developer.download.nvidia.com/compute/redist/jp/v42 tensorflow-gpu --user
则不会有h5py报错的问题。
若jetpack版本为4.1.1
,则:
pip3 install --extra-index-url https://developer.download.nvidia.com/compute/redist/jp/v411 tensorflow-gpu==1.13.0rc0+nv19.2 --user
以上获取的都是最新的tf-gpu支持,若想获取指定版本,可以:
sudo pip3 install --extra-index-url https://developer.download.nvidia.com/compute/redist/jp/v42 tensorflow-gpu==$TF_VERSION+nv$NV_VERSION
其中:
TF_VERSION
:tensorflow版本,比如1.12.0
。
NV_VERSION
:nvidia每月的tensorflow容器版本,比如19.01
。
完整命令类似:
sudo pip3 install --extra-index-url https://developer.download.nvidia.com/compute/redist/jp/v42 tensorflow-gpu==1.12.0+nv19.1
hope u enjoy?????????