tensorflow-haskell/tools/nvidia-docker-wrapper.sh

10 lines
247 B
Bash
Executable File

#!/bin/bash
# This wrapper is necessary to fool 'stack --docker' into
# using nvidia-docker which again will invoke the real
# docker in the end.
#
# Symlink this file as 'docker' somewhere in path
NV_DOCKER='/usr/bin/docker' nvidia-docker "$@"