Jetson Xavier NXとdockerでtensorflow環境の構築

Jetson Xavier NXでtensorflow環境構築の備忘録です

contents

スポンサーリンク

環境

Jetson Xavier NX
ubuntu18.04
docker
python3.x

環境構築手順

1.以下のコマンドでdocker imageを取得

sudo docker pull nvcr.io/nvidia/l4t-tensorflow:r32.4.3-tf1.15-py3

2.以下コマンドでcontainerに入る

sudo docker run -it --rm --runtime nvidia --network host nvcr.io/nvidia/l4t-tensorflow:r32.4.3-tf1.15-py3

3.pytorchのバージョンを確認する

pip3 list

->listにtensorflowがあればOK

スポンサーリンク

関連記事

deeplearningの他フレームワークの環境構築は以下記事で行ってますのでご参考までに
technoxs-stacker.hatenablog.com technoxs-stacker.hatenablog.com

参考

ngc.nvidia.com

感想

ngcのdocker imageをもとにモジュール追加していけば環境構築楽ですね～