docker 環境内に nvidia-driver をインストールします。
ただし、外部バージョンとの不一致は簡単に発生し、バージョンの不一致などのさまざまなエラーが発生します
docker イメージ コンテナーで gpu を使用するには、nVidia の公式パッケージである nvidia-docker を使用することをお勧めします。
1. docker 環境に nvidia ドライバーがないことを確認して、以降の構成でドライバーのバージョンが一致しないようにします。
GPU を使用するイメージを見つけ、そのイメージを入力して、関連するドライバー パッケージを削除します。
sudo apt-get --purge remove "*nvidia*"
2. docker イメージから ubuntu-container-toolkit をダウンロードしてインストールし、docker を再起動します (そうしないと、docker は新しくインストールされたツールキットを見つけることができません)
sudo apt-get install -y nvidia-container-toolkit
sudo systemctl restart docker
3. インストール後、スタートアップ コンテナー スクリプトで docker run --gpus all を使用して、docker イメージで nvidia-smi を使用します。
docker run --gpus all 你要用gpu的镜像名 nvidia-smi