以前に既存のdocker imageを使ってjupyterlabの環境を作りました
technoxs-stacker.hatenablog.com

今後jupyterlab上でpytrorchを動かすこともあるかと思い、以前のpytorch docker imageをもとに
jupyterlabをインストールしたimageを作ってみました

この記事でわかること

pytorchが使えるjupyterlab imageの作り方

1.実行環境

Jetson Xavier NX
ubuntu18.04
docker
python3.x

2.Dockerfileの作成

FROM nvcr.io/nvidia/l4t-ml:r32.5.0-py3
RUN apt-get update && apt-get upgrade -y && apt-get clean
RUN apt-get -y install vim

RUN pip3 install --upgrade pip
RUN pip3 install --ignore-installed PyYAML
RUN pip3 install jupyterlab

ARG USERNAME=user
ARG GROUPNAME=user
ARG UID=1000
ARG GID=1000
ARG PASSWORD=xxxxxxxxx←任意のパスワード
RUN groupadd -g $GID $GROUPNAME && \
    useradd -m -s /bin/bash -u $UID -g $GID -G sudo $USERNAME && \
    echo $USERNAME:$PASSWORD | chpasswd && \
    echo "$USERNAME   ALL=(ALL) NOPASSWD:ALL" >> /etc/sudoers
USER $USERNAME

3.imageの作成

以下コマンドでimageを作成

sudo docker build . -t xxxx←任意のタブ

4.docker containerの起動

sudo docker run -v path/to/mount/:/mount/name/ -it --rm -p port:port --gpus all --name image_name  xxxx

今回使用したオプションのうち、一部をいかに記載します
※詳しくはdockerの公式を参照下さい
-v : マウントしたいディレクトリがあれば指定
-p : コンテナとマシンをつなぐportを指定
--name : コンテナの名前を指定
xxxx: 3で作成したimageを指定

5.jupyterlabの起動

jupyter notebook --port 設定したport番号 --allow-root --ip 0.0.0.0

6.参考

qiita.com

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

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

【jupyter lab】Jetson Xavier NXとdockerでpytorchが使えるjupyterlab環境を構築する

目次