Francesco Gatti
22 lines
562 B
Markdown
22 lines
562 B
Markdown
# 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
|
|
```
|
|
|