mmr/tkDNN

Archived

This repository has been archived on 2026-02-22. You can view files and clone it. You cannot open issues or pull requests or push a commit.

Files

T

History

perseusdg c0e2097397 - CMakeLists.txt opencv cuda contrib autodetect

- Updated Docker to cuda-11.3+cudnn-8.2.1+TensorRT-8.0.34,Ubuntu to 20.04 and OpenCV to 4.5.4
- Updated OpenCV4 to 4.5.4 in install_OpenCV4.sh
- Updated README.md

2021-11-10 12:30:57 +05:30

Dockerfile

docker

2020-07-13 19:50:00 +02:00

Dockerfile.base

- CMakeLists.txt opencv cuda contrib autodetect

2021-11-10 12:30:57 +05:30

README.md

Update README.md

2020-07-13 19:51:09 +02:00

README.md

Use the prebuilt image

# build image
docker build -t tkdnn:build -f Dockerfile .

Build Base Docker image

# make nvidia docker working
# follow this guide: https://github.com/NVIDIA/nvidia-docker

# dowload tensorrt
# from: https://developer.nvidia.com/compute/machine-learning/tensorrt/secure/7.0/7.0.0.11/local_repo/nv-tensorrt-repo-ubuntu1804-cuda10.2-trt7.0.0.11-ga-20191216_1-1_amd64.deb

# build image
docker build -t ceccocats/tkdnn:latest -f Dockerfile.base .

# run image
docker run -ti --gpus all --rm ceccocats/tkdnn:latest bash