Repolode
iot-salzburg

iot-salzburg/gpu-jupyter

再現性の高いディープラーニングに使える、PyTorch、TensorFlow、データサイエンス用ツール群を備えたGPU対応JupyterLabのDockerイメージ。

スター
768
フォーク
238
未解決の Issue
3
最終コミット
3 か月前

これは何?

GPU-Jupyterは、NVIDIAの公式CUDAイメージとJupyter Docker Stacksをベースにした、DockerベースのJupyterLab環境です。PyTorch、TensorFlow、各種データサイエンス用パッケージが含まれているため、CUDAライブラリやフレームワークを手動でインストールせずに、GPU対応のノートブックを実行できます。

このプロジェクトは再現性を重視しています。CUDA、Ubuntu、ライブラリのバージョンを固定したタグ付きイメージが公開されており、ローカルフォルダをコンテナにマウントすれば、ノートブックやデータをセッションをまたいで保持できます。JuliaとRを含むフルビルドから、Pythonのみの軽量ビルドまで、複数のイメージバリアントが用意されています。

複数の方法でカスタマイズできます。Docker Composeでイメージを実行したり、固定のJupyterトークンを設定したり、pipやaptで追加パッケージをインストールしたりできます。また、使用するCUDAベースイメージとJupyter Docker Stacksのコミットを指定して、独自のDockerfileを生成・ビルドすることもできます。

主な機能

  • 01PyTorchとTensorFlowを備えた、NVIDIA CUDA対応のJupyterLab環境。
  • 02実験の再現性を高める、バージョンタグ付きイメージ。
  • 03コンテナ外に作業内容を保持できるローカルフォルダのマウント。
  • 04JuliaとRを追加できる、Pythonのみの軽量版とフル版。
  • 05Docker Composeに対応し、固定のJupyterトークンも設定可能。
  • 06カスタムDockerfileとイメージをビルドする生成スクリプト。

ユースケース

  • ローカルマシンやサーバーのノートブックで、GPUを使ったディープラーニング実験を実行する。
  • 共同研究者との共有や論文公開に向けて、再現可能な研究環境を用意する。
  • ドライバーやライブラリを手動で設定せずに、GPUホストに標準的なデータサイエンス環境を構築する。
  • ベースイメージに独自のパッケージを追加し、チーム専用のイメージを作成する。

クイックスタート

  1. 1

    必要なものをインストールします:NVIDIA GPU ドライバー、CUDA、Docker、NVIDIA Container Toolkit

  2. 2

    Docker で GPU にアクセスできることを確認します

    docker run --rm --gpus all nvidia/cuda:13.0.3-cudnn-runtime-ubuntu24.04 nvidia-smi
  3. 3

    GPU-Jupyter イメージを pull して実行し、表示された token を使って http://localhost:8848 を開きます

    docker run --gpus all -it -p 8848:8888 -v "$(pwd)/data:/home/jovyan/work" -e GRANT_SUDO=yes -e JUPYTER_ENABLE_LAB=yes --user root cschranz/gpu-jupyter:v1.11_cuda-13.0_ubuntu-24.04_python-only

使う前に知っておきたいこと

  • NVIDIA GPU と、インストール済みのドライバー、CUDA、Docker、NVIDIA Container Toolkit が必要です。
  • イメージのサイズが大きいため、初回のダウンロードには時間がかかることがあります。

トピック

よくある質問

gpu-jupyter とは?
再現性の高いディープラーニングに使える、PyTorch、TensorFlow、データサイエンス用ツール群を備えたGPU対応JupyterLabのDockerイメージ。 GPU-Jupyterは、NVIDIAの公式CUDAイメージとJupyter Docker Stacksをベースにした、DockerベースのJupyterLab環境です。PyTorch、TensorFlow、各種データサイエンス用パッケージが含まれているため、CUDAライブラリやフレームワークを手動でインストールせずに、GPU対応のノートブックを実行できます。
gpu-jupyter は無料で使えますか?ライセンスは?
gpu-jupyter は Apache-2.0 ライセンスのオープンソースです。商用製品で使う前にライセンス条件を確認してください。
gpu-jupyter は今もメンテナンスされていますか?
半年以内に更新あり。gpu-jupyter の最終コミットは 2026年8月11日 です。
gpu-jupyter は何に使えますか?
ローカルマシンやサーバーのノートブックで、GPUを使ったディープラーニング実験を実行する。 共同研究者との共有や論文公開に向けて、再現可能な研究環境を用意する。 ドライバーやライブラリを手動で設定せずに、GPUホストに標準的なデータサイエンス環境を構築する。 ベースイメージに独自のパッケージを追加し、チーム専用のイメージを作成する。
gpu-jupyter のインストール方法は?
必要なものをインストールします:NVIDIA GPU ドライバー、CUDA、Docker、NVIDIA Container Toolkit Docker で GPU にアクセスできることを確認します: docker run --rm --gpus all nvidia/cuda:13.0.3-cudnn-runtime-ubuntu24.04 nvidia-smi GPU-Jupyter イメージを pull して実行し、表示された token を使って http://localhost:8848 を開きます: docker run --gpus all -it -p 8848:8888 -v "$(pwd)/data:/home/jovyan/work" -e GRANT_SUDO=yes -e JUPYTER_ENABLE_LAB=yes --user root cschranz/gpu-jupyter:v1.11_cuda-13.0_ubuntu-24.04_python-only
gpu-jupyter はどこで入手できますか?
ソースコードとインストール手順は GitHub にあります:https://github.com/iot-salzburg/gpu-jupyter

概要はプロジェクトの README をもとに AI が生成しています。GitHub の数値は毎日更新されます。

関連プロジェクト