File size: 218 Bytes
c5ca37a
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
SCRIPTPATH="/home/chunyl/research/project/optimus"
IMAGE=chunyl/pytorch-transformers:v2

docker run \
--runtime=nvidia \
-it --rm \
--net host \
--volume $SCRIPTPATH:/workspace \
--interactive --tty $IMAGE /bin/bash