24 lines
498 B
YAML
24 lines
498 B
YAML
|
version: '2'
|
||
|
|
|
||
|
|
services:
|
||
|
|
nvidia-ffmpeg:
|
||
|
|
build: ./
|
||
|
|
image: nvidia-ffmpeg
|
||
|
|
container_name: nvidia-ffmpeg
|
||
|
|
restart: "no"
|
||
|
|
entrypoint:
|
||
|
|
- /bin/sleep
|
||
|
|
- infinity
|
||
|
|
volumes:
|
||
|
|
- /usr/lib/x86_64-linux-gnu/nvidia/current:/cuda:ro
|
||
|
|
- $PWD/folder:/folder
|
||
|
|
environment:
|
||
|
|
- NVIDIA_VISIBLE_DEVICES=all
|
||
|
|
devices:
|
||
|
|
- /dev/nvidia0
|
||
|
|
- /dev/nvidiactl
|
||
|
|
- /dev/nvidia-uvm
|
||
|
|
- /dev/nvidia-uvm-tools
|
||
|
|
cap_add:
|
||
|
|
- IPC_LOCK
|
||
|
|
network_mode: host
|