15 lines
336 B
Bash
Executable file
15 lines
336 B
Bash
Executable file
#!/bin/bash
|
|
|
|
CMD=${@:-/bin/bash}
|
|
NV_VISIBLE_DEVICES=${NVIDIA_VISIBLE_DEVICES:-"all"}
|
|
|
|
|
|
nvidia-docker run --rm -it \
|
|
--net=host \
|
|
--shm-size=1g \
|
|
--ulimit memlock=-1 \
|
|
--ulimit stack=67108864 \
|
|
-e NVIDIA_VISIBLE_DEVICES=$NV_VISIBLE_DEVICES \
|
|
-v $PWD:/workspace/bert \
|
|
-v $PWD/results:/results \
|
|
bert $CMD
|