tfbenchmark

Running tensorflow benchmark with docker

preinstall: Driver & Docker & Nvidia-Docker2

詳細安裝步驟請見Install Nvidia Driver&Docker&NV Docker

Build a tensorflow-benchmark image based on tensorflow/benchmark:nightly-gpu-py3

#至桌面新增資料夾，並在資料夾內下載 Dockerfile製作成image
cd ~/Desktop
mkdir tf_benchmark
cd tf_benchmark
wget https://raw.githubusercontent.com/fatwinnie/tf_benchmark/master/Dockerfile
docker build -t tfbench_nightly .

Run tensorflow benchmark

nvidia-docker run -ti --rm tfbench_nightly

or

docker run --runtime=nvidia -it --rm tfbench_nightly

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