mirror of
https://github.com/pstrueb/piper.git
synced 2026-04-18 14:24:49 +00:00
15 lines
356 B
Bash
Executable File
15 lines
356 B
Bash
Executable File
#!/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 /etc/hostname:/etc/hostname:ro \
|
|
-v /etc/localtime:/etc/localtime:ro \
|
|
piper-train \
|
|
"$@"
|