Docker コンテナーで NVIDIA GPU をパススルーし、これらの Docker コンテナーから NVIDIA GPU で CUDA プログラムを実行できます。これは、AI(人工知能)の学習に非常に役立つ機能です。 Docker コンテナーで AI コード (Tensorflow など) を実行できると、時間を大幅に節約できます。コンピューターの CUDA バージョンを変更する必要なく、さまざまな CUDA バージョンで AI コードを試すこともできます。
コンピューターで NVIDIA ドライバーと CUDA バージョンをいじると、NVIDIA ドライバーが機能しない、またはブラック/ブルー スクリーンが表示されるなどの問題が発生する可能性があります。そのため、システムをそのままにしておき、Docker コンテナーのような隔離された環境で変更を加えることは常に良い考えです。これにより、メイン コンピューターを (不要な開発ツールから) クリーンに保つことができます。
この記事では、Docker コンテナーからコンピューターの NVIDIA GPU にアクセスし、NVIDIA GPU で CUDA プログラムを実行できるように、Ubuntu 22.04 LTS で Docker CE と NVIDIA Docker をセットアップする方法を紹介します。
目次:
- 公式の NVIDIA GPU ドライバーが Ubuntu 22.04 にインストールされているかどうかを確認する
- Ubuntu 22.04 への Docker CE 依存関係のインストール
- Ubuntu 22.04 への Docker CE GPG キーのインストール
- Ubuntu 22.04 への Docker CE リポジトリのインストール
- Ubuntu 22.04 への Docker CE のインストール
- Ubuntu 22.04 LTS ログイン ユーザーを docker グループに追加する
- Docker CE が Ubuntu 22.04 に正しくインストールされたかどうかを確認する
- Ubuntu 22.04 への NVIDIA Container Toolkit GPG キーのインストール
- Ubuntu 22.04 への NVIDIA Container Toolkit リポジトリのインストール
- Ubuntu 22.04 への nvidia-docker ドライバーのインストール
- NVIDIA GPU が Ubuntu 22.04 の Docker コンテナーからアクセス可能かどうかを確認する
- 結論
- 参考文献
公式の NVIDIA GPU ドライバーが Ubuntu 22.04 LTS にインストールされているかどうかを確認します。
開始する前に、次のコマンドを実行して、コンピューターに NVIDIA GPU がインストールされていることを確認します。
$ lspci | | egrep -私 「VGA|3D|ディスプレイ」
この場合、コンピューターに NVIDIA GTX 1050 Ti GPU がインストールされています。ほとんどの場合、コンピューターには別の NVIDIA GPU がインストールされています。
また、次のコマンドを使用して、公式の NVIDIA ドライバーが Ubuntu 22.04 オペレーティング システムにインストールされていることを確認してください。
$ lsmod | | グリップ NVIDIA
公式の NVIDIA ドライバーが Ubuntu 22.04 オペレーティング システムにインストールされている場合、次の出力が表示されます。
また、公式の NVIDIA GPU ドライバーが次のコマンドで動作しているかどうかを確認します。
公式の NVIDIA GPU ドライバーが動作している場合は、次の出力が表示されます。ご覧のとおり、公式の NVIDIA GPU ドライバー バージョン 525.78.01 が Ubuntu 22.04 マシンにインストールされています。
公式の NVIDIA GPU ドライバーが Ubuntu 22.04 マシンにインストールされておらず、サポートが必要な場合は、記事を確認してください。 .
Ubuntu 22.04 LTS への Docker CE 依存関係のインストール:
Ubuntu 22.04 に Docker CE をインストールする前に、必要な Docker CE 依存パッケージを Ubuntu 22.04 にインストールする必要があります。
最初に、次のコマンドで APT パッケージ リポジトリ キャッシュを更新します。
$ 須藤 適切な更新
APT パッケージ リポジトリ キャッシュを更新する必要があります。
Docker CE の必要な依存パッケージをインストールするには、次のコマンドを実行します。
インストールを確認するには、 を押します。 と を押します。 <入力> .
必要な Docker CE 依存パッケージをインストールする必要があります。
Ubuntu 22.04 LTS への Docker CE GPG キーのインストール:
このセクションでは、Docker CE パッケージ リポジトリの GPG キーを Ubuntu 22.04 にインストールする方法を説明します。
まず、新しいフォルダを作成します /etc/apt/キーリング 次のコマンドを使用します。
$ 須藤 mkdir -p / 等 / 適切な / キーリング
Docker CE パッケージ リポジトリの GPG キーを Ubuntu 22.04 にインストールするには、次のコマンドを実行します。
$ カール -fsSL https: /// ダウンロード.docker.com / Linux / Ubuntu / gpg | | 須藤 gpg --dearmor -O / 等 / 適切な / キーリング / docker.gpg
Ubuntu 22.04 LTS への Docker CE リポジトリのインストール:
Docker CE パッケージ リポジトリを Ubuntu 22.04 にインストールするには、次のコマンドを実行します。
$ エコー 'デブ[アーチ= $(dpkg --print-architecture) signed-by=/etc/apt/keyrings/docker.gpg] https://download.docker.com/linux/ubuntu $(lsb_release -cs) 安定' | | 須藤 ティー / 等 / 適切な / ソース.リスト.d / docker.list > / 開発者 / ヌル
変更を有効にするには、APT パッケージ リポジトリ キャッシュを更新します。
$ 須藤 適切な更新
Ubuntu 22.04 LTS への Docker CE のインストール:
Docker CE の最新バージョンを Ubuntu 22.04 にインストールするには、次のコマンドを実行します。
$ 須藤 apt-get インストール docker-ce docker-ce-cli containerd.io docker-compose-plugin
インストールを確認するには、 を押します。 と を押します。 <入力> .
Docker CE と必要な依存パッケージがダウンロードされています。完了するまでしばらく時間がかかります。
Docker CE と必要な依存パッケージがインストールされています。完了するまでしばらく時間がかかります。
この時点で、Docker CE と必要な依存パッケージをインストールする必要があります。
Ubuntu 22.04 LTS ログイン ユーザーを docker グループに追加します。
sudo を使用したり、root ユーザーとしてログインしたりせずに、Docker コンテナーを作成して管理するには、ログイン ユーザーを ドッカー グループ。
Ubuntu 22.04 のログイン ユーザーを ドッカー グループで、次のコマンドを実行します。
$ 須藤 ユーザーモード -aG ドッカー $ ( 私は誰 )
変更を有効にするには、次のコマンドでコンピューターを再起動します。
$ 須藤 リブート
Docker CE が Ubuntu 22.04 LTS に正しくインストールされているかどうかの確認:
コンピューターが起動したら、次のコマンドを実行して、スーパーユーザー権限なしで Docker にアクセスできることを確認します。
$ ドッカーのバージョン
すべてが機能する場合、次の出力が表示されます。
ご覧のとおり、Docker バージョン 20.10.23 を実行しています。これは、この記事の執筆時点での最新バージョンの Docker CE です。
Ubuntu 22.04 への NVIDIA Container Toolkit GPG キーのインストール:
このセクションでは、NVIDIA Container Toolkit パッケージ リポジトリの GPG キーを Ubuntu 22.04 にインストールする方法を説明します。
NVIDIA Container Toolkit パッケージ リポジトリの GPG キーを Ubuntu 22.04 にインストールするには、次のコマンドを実行します。
$ カール -fsSL https: /// nvidia.github.io / libnvidia-コンテナ / gpgkey | | 須藤 gpg --dearmor -O / ユーザー / 共有 / キーリング / nvidia-container-toolkit-keyring.gpg
Ubuntu 22.04 LTS への NVIDIA Container Toolkit リポジトリのインストール:
このセクションでは、NVIDIA Container Toolkit パッケージ リポジトリを Ubuntu 22.04 にインストールする方法を紹介します。
まず、新しい APT ソース ファイルを作成します。 nvidia-container-toolkit.list の中に /etc/apt/sources.list.d/ ディレクトリは次のとおりです。
$ 須藤 ナノ / 等 / 適切な / ソース.リスト.d / nvidia-container-toolkit.list
の中に nvidia-container-toolkit.list ファイルに次の行を追加して押します
Ubuntu 20.04 LTS を使用している場合は、次の行を nvidia-container-toolkit.list 代わりにファイルを使用すると、他のすべてが変更されずに機能するはずです。
Ubuntu 18.04 LTS を使用している場合は、次の行を nvidia-container-toolkit.list 代わりにファイルを使用すると、他のすべてが変更されずに機能するはずです。
変更を有効にするには、次のコマンドで APT パッケージ リポジトリ キャッシュを更新します。
Ubuntu 22.04 LTS への nvidia-docker ドライバーのインストール:
Ubuntu 22.04 に NVIDIA Docker ドライバーをインストールするには、次のコマンドを実行します。
$ 須藤 適切な インストール nvidia-docker2
インストールを確認するには、 を押します。 と を押します。 <入力> .
NVIDIA docker ドライバーがインストールされている必要があります。
変更を有効にするには、次のコマンドでコンピューターを再起動します。
NVIDIA GPU が Ubuntu 22.04 LTS の Docker コンテナーからアクセス可能かどうかを確認します。
このセクションでは、 NVIDIA CUDA Docker コンテナー コンテナーがコンピューターから NVIDIA GPU にアクセスできることを確認します。
Ubuntu 20.04 LTS に基づいて NVIDIA CUDA 12 Docker コンテナーを作成し、 nvidia-smi コンピューターから NVIDIA GPU にアクセスできるかどうかを確認するために作成されたら、次のコマンドを実行します。
$ ドッカー実行 --rm -- gpu すべてのNVIDIA / cuda:12.0.0-base-ubuntu20.04 nvidia-smi
Docker は nvidia/cuda:12.0.0-base-ubuntu20.04 イメージは Docker Hub から。完了するまでしばらく時間がかかります。
NVIDIA CUDA Docker イメージがプルされ、コンテナーが作成されると、 nvidia-smi コマンドが実行され、コンソールに出力が表示されます (下のスクリーンショットを参照)。
Docker コンテナは NVIDIA GPU ドライバー 525.78.01 [1] そしてその CUDA バージョン 12.0 [2] .同様の出力が表示された場合、Docker コンテナーはコンピューターの NVIDIA GPU にアクセスできます。
古いバージョンの CUDA を使用する必要がある場合は、以下を確認してください。 .
結論:
この記事では、Ubuntu 22.04 で Docker CE パッケージ リポジトリをセットアップする方法を説明しました。また、最新バージョンの Docker CE を Ubuntu 22.04 にインストールする方法も示しました。 NVIDIA Container Toolkit パッケージ リポジトリを Ubuntu 22.04 にインストールする方法と、NVIDIA Docker ドライバーを Ubuntu 22.04 にインストールする方法を説明しました。最後に、Docker コンテナーからコンピューターの NVIDIA GPU にアクセスする方法を示しました。