File size: 394 Bytes
99f9967
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
#!/usr/bin/env bash

# Follow instructions here: https://docs.docker.com/config/containers/resource_constraints/#access-an-nvidia-gpu
docker run \
  -it \
  --gpus all \
  -w "$PWD" \
  --user "$(id -u):$(id -g)" \
  --ipc=host \
  -v "${HOME}:${HOME}" \
  -v /media/cache:/media/cache:ro \
  -v /etc/hostname:/etc/hostname:ro \
  -v /etc/localtime:/etc/localtime:ro \
  larynx2-train \
  "$@"