Commit ·
f8bb793
1
Parent(s): 9a1798b
transcribe container now properly uses GPU
Browse files- compose.yml +4 -2
compose.yml
CHANGED
|
@@ -1,6 +1,6 @@
|
|
| 1 |
services:
|
| 2 |
transcribe:
|
| 3 |
-
image: linuxserver/faster-whisper:
|
| 4 |
container_name: transcribe
|
| 5 |
environment:
|
| 6 |
- NVIDIA_VISIBLE_DEVICES=all # GPU Support
|
|
@@ -8,7 +8,9 @@ services:
|
|
| 8 |
runtime: nvidia # GPU Support
|
| 9 |
volumes:
|
| 10 |
- "./:/training:rw"
|
| 11 |
-
|
|
|
|
|
|
|
| 12 |
train:
|
| 13 |
build: .
|
| 14 |
container_name: training
|
|
|
|
| 1 |
services:
|
| 2 |
transcribe:
|
| 3 |
+
image: linuxserver/faster-whisper:2.4.0-gpu
|
| 4 |
container_name: transcribe
|
| 5 |
environment:
|
| 6 |
- NVIDIA_VISIBLE_DEVICES=all # GPU Support
|
|
|
|
| 8 |
runtime: nvidia # GPU Support
|
| 9 |
volumes:
|
| 10 |
- "./:/training:rw"
|
| 11 |
+
- "./cudnn_fix:/config/.bashrc:ro"
|
| 12 |
+
working_dir: /training
|
| 13 |
+
|
| 14 |
train:
|
| 15 |
build: .
|
| 16 |
container_name: training
|