Video2X in Colab crashing with RTX6000 GPU

[Goldbeibei]

b0942103@ubuntu:~$ docker run -it --rm -v /home/b0942103:/host ghcr.io/k4yt3x/video2x:5.0.0-beta5-slim-alpine -i input.mp4 -o output.mp4 -p3 upscale -h 720 -a waifu2x -n3
10:40:59.996268 | INFO | Video2X 5.0.0-beta5
10:40:59.996574 | INFO | Copyright (C) 2018-2022 K4YT3X and contributors.
10:40:59.996721 | INFO | Reading input video information
10:41:00.098644 | INFO | Starting video decoder
10:41:00.105011 | INFO | Starting video encoder
10:41:00.116592 | INFO | Upscaler process 0 initiating
10:41:00.119264 | INFO | Upscaler process 1 initiating
10:41:00.120508 | INFO | Upscaler process 2 initiating
vkEnumeratePhysicalDevices failed -3
vkEnumeratePhysicalDevices failed -3
vkEnumeratePhysicalDevices failed -3
Input #0, mov,mp4,m4a,3gp,3g2,mj2, from 'input.mp4':
Metadata:
major_brand : mp42
minor_version : 0
compatible_brands: isommp42
creation_time : 2020-10-15T23:31:39.000000Z
encoder : Google
Duration: 00:11:07.32, start: 0.000000, bitrate: 374 kb/s
Stream #0:0(und): Video: h264 (Main) (avc1 / 0x31637661), yuv420p(tv, bt709), 640x360 [SAR 1:1 DAR 16:9], 275 kb/s, 30 fps, 30 tbr, 15360 tbn, 60 tbc (default)
Metadata:
creation_time : 2020-10-15T23:31:39.000000Z
handler_name : ISO Media file produced by Google Inc. Created on: 10/15/2020.
vendor_id : [0][0][0][0]
Stream #0:1(und): Audio: aac (LC) (mp4a / 0x6134706D), 44100 Hz, stereo, fltp, 96 kb/s (default)
Metadata:
creation_time : 2020-10-15T23:31:39.000000Z
handler_name : ISO Media file produced by Google Inc. Created on: 10/15/2020.
vendor_id : [0][0][0][0]
Stream mapping:
Stream #0:0 -> #0:0 (h264 (native) -> rawvideo (native))
Output #0, rawvideo, to 'pipe:1':
Metadata:
major_brand : mp42
minor_version : 0
compatible_brands: isommp42
encoder : Lavf58.76.100
Stream #0:0(und): Video: rawvideo (RGB[24] / 0x18424752), rgb24(pc, gbr/bt709/bt709, progressive), 640x360 [SAR 1:1 DAR 16:9], q=2-31, 165888 kb/s, 30 fps, 30 tbn (default)
Metadata:
creation_time : 2020-10-15T23:31:39.000000Z
handler_name : ISO Media file produced by Google Inc. Created on: 10/15/2020.
vendor_id : [0][0][0][0]
encoder : Lavc58.134.100 rawvideo

10:41:01.118628 | ERROR | process died unexpectedly
Traceback (most recent call last):

File "/usr/local/lib/python3.10/runpy.py", line 196, in _run_module_as_main
return _run_code(code, main_globals, None,
| | -> {'name': 'main', 'doc': '\nCopyright (C) 2018-2022 K4YT3X and contributors.\n\nThis program is free software: you...
| -> <code object at 0x7f70f99cf1b0, file "/usr/local/lib/python3.10/site-packages/video2x/main.py", line 1>
-> <function _run_code at 0x7f7105496170>
File "/usr/local/lib/python3.10/runpy.py", line 86, in _run_code
exec(code, run_globals)
| -> {'name': 'main', 'doc': '\nCopyright (C) 2018-2022 K4YT3X and contributors.\n\nThis program is free software: you...
-> <code object at 0x7f70f99cf1b0, file "/usr/local/lib/python3.10/site-packages/video2x/main.py", line 1>
File "/usr/local/lib/python3.10/site-packages/video2x/main.py", line 30, in
sys.exit(main())
| | -> <function main at 0x7f70f9804430>
| ->
-> <module 'sys' (built-in)>
File "/usr/local/lib/python3.10/site-packages/video2x/video2x.py", line 612, in main
video2x.upscale(
| -> <function Video2X.upscale at 0x7f70f9804310>
-> <video2x.video2x.Video2X object at 0x7f70f9c971c0>
File "/usr/local/lib/python3.10/site-packages/video2x/video2x.py", line 412, in upscale
self._run(
| -> <function Video2X._run at 0x7f70f9804280>
-> <video2x.video2x.Video2X object at 0x7f70f9c971c0>

File "/usr/local/lib/python3.10/site-packages/video2x/video2x.py", line 308, in _run
raise Exception("process died unexpectedly")

Exception: process died unexpectedly
10:41:01.136182 | INFO | Stopping processor processes
10:41:01.136949 | INFO | Stopping decoder and encoder threads
10:41:01.236209 | DEBUG | Decoding queue depleted
10:41:01.238150 | DEBUG | Encoding queue depleted
10:41:01.615044 | INFO | Encoder thread exiting
10:41:01.616146 | INFO | Decoder thread exiting
10:41:01.617405 | ERROR | process died unexpectedly
Traceback (most recent call last):

File "/usr/local/lib/python3.10/runpy.py", line 196, in _run_module_as_main
return _run_code(code, main_globals, None,
| | -> {'name': 'main', 'doc': '\nCopyright (C) 2018-2022 K4YT3X and contributors.\n\nThis program is free software: you...
| -> <code object at 0x7f70f99cf1b0, file "/usr/local/lib/python3.10/site-packages/video2x/main.py", line 1>
-> <function _run_code at 0x7f7105496170>
File "/usr/local/lib/python3.10/runpy.py", line 86, in _run_code
exec(code, run_globals)
| -> {'name': 'main', 'doc': '\nCopyright (C) 2018-2022 K4YT3X and contributors.\n\nThis program is free software: you...
-> <code object at 0x7f70f99cf1b0, file "/usr/local/lib/python3.10/site-packages/video2x/main.py", line 1>
File "/usr/local/lib/python3.10/site-packages/video2x/main.py", line 30, in
sys.exit(main())
| | -> <function main at 0x7f70f9804430>
| ->
-> <module 'sys' (built-in)>

File "/usr/local/lib/python3.10/site-packages/video2x/video2x.py", line 612, in main
video2x.upscale(
| -> <function Video2X.upscale at 0x7f70f9804310>
-> <video2x.video2x.Video2X object at 0x7f70f9c971c0>
File "/usr/local/lib/python3.10/site-packages/video2x/video2x.py", line 412, in upscale
self._run(
| -> <function Video2X._run at 0x7f70f9804280>
-> <video2x.video2x.Video2X object at 0x7f70f9c971c0>
File "/usr/local/lib/python3.10/site-packages/video2x/video2x.py", line 375, in _run
raise exception[0]
-> [Exception('process died unexpectedly')]
File "/usr/local/lib/python3.10/site-packages/video2x/video2x.py", line 308, in _run
raise Exception("process died unexpectedly")

Exception: process died unexpectedly

[k4yt3x]

slim-alpine only contains Intel and AMD drivers, which is why it's a lot smaller. For NVIDIA GPUs, use the full version.