tfbenchmark

Running tensorflow benchmark and nbody with docker

nbody-docker

Build a nbody image based on nvidia/cudagl:tag

wget https://raw.githubusercontent.com/hao3924/tfbenchmark/nbody-docker/Dockerfile
docker build -t dqa4/nbody:[cuda_version] [PATH of Dockerfile]
#e.g.  docker build -t dqa4/nbody:cuda-10.1 .

Run nbody benchmark

nvidia-docker run -it --rm dqa4/nbody:[cuda_version] ./nbody -benchmark
#e.g.  nvidia-docker run -it --rm dqa4/nbody:cuda-10.1 ./nbody -benchmark

or

docker run --runtime=nvidia -it --rm dqa4/nbody:[cuda_version] ./nbody -benchmark

FAQ

Q: Premission denied when docker build, docker run?

A: Add sudo before docker e.g. sudo docker build or change your user as docker administrator by using sudo usermod -aG docker $USER