1
0
mirror of https://github.com/tensorflow/haskell.git synced 2024-06-02 19:13:34 +02:00
tensorflow-haskell/tools/nvidia-docker-wrapper.sh

10 lines
247 B
Bash
Raw Normal View History

#!/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 "$@"