dGPU Setup for Ubuntu
次のコンポーネントインストールが必要です。ただし、必ず以下のバージョンでの組み合わせでありません。高い以下のバージョンでも交換性があります。
- Ubuntu 18.04
- GStreamer 1.14.3(1.14.5)
- NVIDIA driver 460.32(460.91.03)
- CUDA 11.1
- TensorRT 7.2.X
依存関係インストール
注意:一つpackageインストール完了したら、必ず以下のコマンドで関連packageを更新してください。
sudo apt update
sudo apt upgrade
gstreamer
NVIDIA driver
460.32インストールとPCが画面が表示できなくなります。
使用しているGPUの確認
lspci | grep -i nvidia
現状入っているCUDA,nvidia-driverの確認
dpkg -l | grep nvidia
dpkg -l | grep cuda
現状入っているCUDA nvidiaドライバの削除(ない場合スキップ)
sudo apt-get --purge remove nvidia-*
sudo apt-get --purge remove cuda-*
nvidia-driver のインストール
以下で推奨のドライバを確認
ubuntu-drivers devices
インストール(nvidia-driver-460の場合) 460.91.03
がインストールされる
sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt update
sudo apt install nvidia-driver-460
再起動
sudo reboot
NVIDIA Driverのバージョンと対応CUDAバージョン確認
nvidia-smi
CUDAのインストール
CUDAインストール
CUDA11.1インストール
現在NVIDIA Driver対応しているCUDAをremoveしてからCUDA11.1インストールすることが必要です。(もしくは同時存在、方法調査中)
CUDAアンインストール [*
]はバージョン指定
sudo apt-get --purge remove cuda-*
CUDA11.1インストールは上記と同じです。
インストール後の確認
sudo apt list | grep cuda
cuda11.1
の後ろにnow
が付いていることは、11.1がインストールされました。
TensorRTのインストール
手順はDeepstream SDK サイト
librdkafka インストール
依存インストール
git
sudo apt install git
sudo apt update
sudo apt upgrade
python
sudo apt install pyhton
sudo apt update
sudo apt upgrade
手順はDeepstream SDK サイト
Install the DeepStream SDK
手順はDeepstream SDK サイト
https://docs.nvidia.com/metropolis/deepstream/dev-guide/text/DS_Quickstart.html
確認
deepstream関連バージョン確認
cudaruntimeのバージョン確認が11.1になっていること
deepstream-app --version-all
deepstreamappサンプル
deepstream-app -c /opt/nvidia/deepstream/deepstream-5.1/samples/configs/deepstream-app/source4_1080p_dec_infer-resnet_tracker_sgie_tiled_display_int8.txt
gstreamerサンプル
sample.jpg自分で準備
gst-launch-1.0 filesrc location=sample.jpg ! jpegdec ! imagefreeze ! autovideosink