neural-style / Makefile
dogeplusplus's picture
Converting to tensorflow serving gpu container.
c71df2a
raw
history blame
187 Bytes
start: style
docker run --gpus all -p 8500:8500 --mount type=bind,source=/home/albert/github/neural-style/style,target=/models/style -e MODEL_NAME=style -t tensorflow/serving:latest-gpu