docker_python.sh

#!/usr/bin/env bash

if hash nvidia-docker 2>/dev/null; then
  cmd=nvidia-docker
else
  cmd=docker
fi

NV_GPU=$1 ${cmd} run --rm -v `pwd`:/home/user/deep_rl --entrypoint '/bin/bash' deep_rl:v1.5 -c "OMP_NUM_THREADS=1 python $2"