Setup a machine learning environment with Jetson Xavier NX and docker

It is a memorandum on setting up a machine learning environment with Jetson Xavier NX.

contents

スポンサーリンク

requirement

Jetson Xavier NX
ubuntu18.04
docker
python3.x

procedure

1.getting docker image

sudo docker pull nvcr.io/nvidia/l4t-ml:r32.5.0-py3

2.enter in a container

sudo docker run -it --rm --runtime nvidia --network host nvcr.io/nvidia/l4t-ml:r32.5.0-py3

3.check the version of modules

pip3 list

->if TensorFlow, PyTorch, scikit-learn, scipy and Pandas are in this list, It's OK

reference

catalog.ngc.nvidia.com

スポンサーリンク