Steve Zaretti
2485106380
root@2a3891735f20:~/tkDNN/build# ./test_yolo4_512 mkdir: cannot create directory 'yolo4_512': File exists sh: 1: curl: not found unzip: cannot find or open yolo4_512/weights.zip, yolo4_512/weights.zip.zip or yolo4_512/weights.zip.ZIP. rm: cannot remove 'yolo4_512/weights.zip': No such file or directory As we install git here instead on https://github.com/ceccocats/tkDNN/blob/master/docker/Dockerfile i guess we can add curl here too.
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