hsh2438/tensorrt

tensorrt

Upgrade Nvidia Driver

• general system upgrade

sudo apt update && sudo apt upgrade

• check available versions

ubuntu-drivers list

• upgrade nvidia driver

sudo apt install nvidia-driver-VERSION_NUMBER

• reboot

Purge Nvidia Driver

sudo apt --purge autoremove nvidia*

TensorRT docker image download

https://ngc.nvidia.com/catalog/containers/nvidia:tensorrt

docker pull nvcr.io/nvidia/tensorrt:20.03-py3 # nvidia driver version should be higher than 440

There is dependency on nvidia driver versions of host. Check dependency from following link. https://docs.nvidia.com/deeplearning/frameworks/support-matrix/index.html

TensorRT container run

docker run --gpus all -it nvcr.io/nvidia/tensorrt:20.03-py3