こすたろーんエンジニアの試行錯誤部屋

作成物の備忘録を書いていきますー

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

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

contents

スポンサーリンク

環境

Jetson Xavier NX
ubuntu18.04
docker
python3.x

環境構築手順

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

sudo docker pull nvcr.io/nvidia/l4t-pytorch:r32.4.4-pth1.6-py3

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

sudo docker run -it --rm --runtime nvidia --network host nvcr.io/nvidia/l4t-pytorch:r32.4.4-pth1.6-py3

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

python3
import torch
print(torch.__version__)

deeplearning関連記事

technoxs-stacker.hatenablog.com technoxs-stacker.hatenablog.com technoxs-stacker.hatenablog.com

スポンサーリンク

参考

ngc.nvidia.com

感想

ngc&dockerがあるとめんどうな環境構築もラクチン~