2017-09-06 04:16:49 +02:00
|
|
|
#! /bin/bash
|
2018-02-23 01:21:27 +01:00
|
|
|
# Copyright (c) 2017-2018, NVIDIA CORPORATION. All rights reserved.
|
2017-09-06 04:16:49 +02:00
|
|
|
|
|
|
|
NV_DOCKER=${NV_DOCKER:-"docker"}
|
|
|
|
|
2018-02-23 01:21:27 +01:00
|
|
|
DOCKER_ARGS=()
|
|
|
|
NV_DOCKER_ARGS=()
|
2017-09-06 04:16:49 +02:00
|
|
|
while [ $# -gt 0 ]; do
|
|
|
|
arg=$1
|
|
|
|
shift
|
2018-02-23 01:21:27 +01:00
|
|
|
DOCKER_ARGS+=("$arg")
|
2017-09-06 04:16:49 +02:00
|
|
|
case $arg in
|
|
|
|
run|create)
|
2018-02-23 01:21:27 +01:00
|
|
|
NV_DOCKER_ARGS+=("--runtime=nvidia")
|
|
|
|
if [ ! -z "${NV_GPU}" ]; then
|
|
|
|
NV_DOCKER_ARGS+=(-e NVIDIA_VISIBLE_DEVICES="${NV_GPU// /,}")
|
2017-09-06 04:16:49 +02:00
|
|
|
fi
|
|
|
|
break
|
|
|
|
;;
|
|
|
|
version)
|
2019-08-12 23:42:31 +02:00
|
|
|
printf "NVIDIA Docker: @VERSION@\n"
|
2017-09-06 04:16:49 +02:00
|
|
|
break
|
|
|
|
;;
|
2017-12-20 05:13:11 +01:00
|
|
|
--)
|
|
|
|
break
|
|
|
|
;;
|
2017-09-06 04:16:49 +02:00
|
|
|
esac
|
|
|
|
done
|
|
|
|
|
|
|
|
if [ ! -z $NV_DEBUG ]; then
|
|
|
|
set -x
|
|
|
|
fi
|
|
|
|
|
2018-02-23 01:21:27 +01:00
|
|
|
exec $NV_DOCKER "${DOCKER_ARGS[@]}" "${NV_DOCKER_ARGS[@]}" "$@"
|