Jetson Nano上のコンテナでpytorch使えるまでメモ（2023/08）

JetPack4.6.1をインストール

SDカードに入れるイメージは以下のリンクにあります。
https://developer.nvidia.com/jetpack-sdk-464

SDカードを焼くのは私はbalenaEtcherを使っていますが好きなのを使えば良いと思います。

nvidia-containerとnvidia-jetpackインストール

sudo apt dist-upgrade
sudo apt update
sudo apt install nvidia-container
sudo apt install nvidia-jetpack

コンテナ立ち上げる

sudo docker pull nvcr.io/nvidia/l4t-ml:r32.5.0-py3
sudo docker run -it --gpus all nvcr.io/nvidia/l4t-ml:r32.5.0-py3

コンテナ

python3
>>> import torch
>>> torch.cuda.is_available()
True