dotNet-Spider
commented
2 weeks ago If you have Nvidia GPU, first try to run this like:
docker run -p 8080:8080 --runtime=nvidia --gpus all facepoke
the --runtime=nvidia --gpus all part is important.
If this doesn't help, login to your WSL2 Linux machine and check if this command shows you your's GPU.
nvidia-smi
Maybe you are missing packages in your WSL2 Linux:
-
NVIDIA Container Toolkit https://docs.nvidia.com/datacenter/cloud-native/container-toolkit/latest/install-guide.html#installing-the-nvidia-container-toolkit
and then run like
docker run -p 8080:8080 --runtime=nvidia --gpus all facepoke
tiandaoyuxi
D:\FacePoke>docker run -p 8080:8080 facepoke
========== == CUDA ==
CUDA Version 12.4.0
Container image Copyright (c) 2016-2023, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
This container image and its contents are governed by the NVIDIA Deep Learning Container License. By pulling and using the container, you accept the terms and conditions of this license: https://developer.nvidia.com/ngc/nvidia-deep-learning-container-license
A copy of this license is made available in this container at /NGC-DL-CONTAINER-LICENSE for your convenience.
WARNING: The NVIDIA Driver was not detected. GPU functionality will not be available. Use the NVIDIA Container Toolkit to start this container with GPU support; see https://docs.nvidia.com/datacenter/cloud-native/ .
2024-10-09 05:31:51,658 - loader - INFO - 🚀 Setting up storage directories... 2024-10-09 05:31:51,658 - loader - INFO - Directory created: /tmp/data/models 2024-10-09 05:31:51,658 - loader - INFO - ✅ Storage directories setup completed. 2024-10-09 05:31:51,746 - main - INFO - Starting FacePoke application 2024-10-09 05:31:51,746 - main - INFO - Initializing application... 2024-10-09 05:31:51,746 - loader - INFO - 🚀 Starting model initialization... 2024-10-09 05:31:51,746 - loader - INFO - 🔎 Looking for models... 2024-10-09 05:31:51,746 - loader - INFO - ⏳ Downloading jbilcke-hf/model-cocktail/dwpose/dw-ll_ucoco_384.pth 2024-10-09 05:31:51,747 - loader - INFO - ⏳ Downloading jbilcke-hf/model-cocktail/face-detector/s3fd-619a316812.pth 2024-10-09 05:31:51,747 - loader - INFO - ⏳ Downloading jbilcke-hf/model-cocktail/liveportrait/spade_generator.pth 2024-10-09 05:31:51,748 - loader - INFO - ⏳ Downloading jbilcke-hf/model-cocktail/liveportrait/warping_module.pth 2024-10-09 05:31:51,749 - loader - INFO - ⏳ Downloading jbilcke-hf/model-cocktail/liveportrait/motion_extractor.pth 2024-10-09 05:31:51,750 - loader - INFO - ⏳ Downloading jbilcke-hf/model-cocktail/liveportrait/stitching_retargeting_module.pth 2024-10-09 05:31:51,752 - loader - INFO - ⏳ Downloading jbilcke-hf/model-cocktail/liveportrait/appearance_feature_extractor.pth 2024-10-09 05:31:51,754 - loader - INFO - ⏳ Downloading jbilcke-hf/model-cocktail/liveportrait/landmark.onnx 2024-10-09 05:31:51,756 - loader - INFO - ⏳ Downloading jbilcke-hf/model-cocktail/insightface/models/buffalo_l.zip 2024-10-09 05:31:51,756 - loader - INFO - ⏳ Downloading jbilcke-hf/model-cocktail/insightface/buffalo_l/det_10g.onnx 2024-10-09 05:31:51,757 - loader - INFO - ⏳ Downloading jbilcke-hf/model-cocktail/insightface/buffalo_l/2d106det.onnx 2024-10-09 05:31:51,757 - loader - INFO - ⏳ Downloading jbilcke-hf/model-cocktail/sd-vae-ft-mse/diffusion_pytorch_model.bin 2024-10-09 05:31:51,759 - loader - INFO - ⏳ Downloading jbilcke-hf/model-cocktail/sd-vae-ft-mse/diffusion_pytorch_model.safetensors 2024-10-09 05:31:51,761 - loader - INFO - ⏳ Downloading jbilcke-hf/model-cocktail/sd-vae-ft-mse/config.json 2024-10-09 05:31:53,160 - loader - INFO - ✅ Downloaded sd-vae-ft-mse/config.json 2024-10-09 05:31:58,040 - loader - INFO - ✅ Downloaded liveportrait/stitching_retargeting_module.pth 2024-10-09 05:31:58,405 - loader - INFO - ✅ Downloaded liveportrait/appearance_feature_extractor.pth 2024-10-09 05:32:03,873 - loader - INFO - ✅ Downloaded insightface/buffalo_l/2d106det.onnx 2024-10-09 05:32:10,918 - loader - INFO - ✅ Downloaded insightface/buffalo_l/det_10g.onnx 2024-10-09 05:33:47,124 - loader - INFO - ✅ Downloaded liveportrait/motion_extractor.pth 2024-10-09 05:33:49,321 - loader - INFO - ✅ Downloaded face-detector/s3fd-619a316812.pth 2024-10-09 05:34:14,391 - loader - INFO - ✅ Downloaded liveportrait/landmark.onnx 2024-10-09 05:34:55,393 - loader - INFO - ✅ Downloaded liveportrait/spade_generator.pth 2024-10-09 05:35:00,544 - loader - INFO - ✅ Downloaded liveportrait/warping_module.pth 2024-10-09 05:35:35,501 - loader - INFO - ✅ Downloaded insightface/models/buffalo_l.zip 2024-10-09 05:35:50,633 - loader - INFO - ✅ Downloaded sd-vae-ft-mse/diffusion_pytorch_model.safetensors 2024-10-09 05:35:56,530 - loader - INFO - ✅ Downloaded sd-vae-ft-mse/diffusion_pytorch_model.bin 2024-10-09 05:36:14,246 - loader - INFO - ✅ Downloaded dwpose/dw-ll_ucoco_384.pth 2024-10-09 05:36:14,246 - loader - INFO - ✅ All models are available 2024-10-09 05:36:15,708 - loader - INFO - ⏳ Loading LivePortrait models... 2024-10-09 05:36:15,719 - main - ERROR - 🚨 Error during application initialization: Found no NVIDIA driver on your system. Please check that you have an NVIDIA GPU and installed a driver from http://www.nvidia.com/Download/index.aspx 2024-10-09 05:36:15,719 - main - ERROR - Full traceback: Traceback (most recent call last): File "/home/user/app/app.py", line 114, in initialize_app live_portrait = await initialize_models() File "/home/user/app/loader.py", line 157, in initialize_models live_portrait = await loader.load_live_portrait() File "/home/user/app/loader.py", line 131, in load_live_portrait live_portrait_pipeline = await asyncio.to_thread( File "/usr/lib/python3.10/asyncio/threads.py", line 25, in to_thread return await loop.run_in_executor(None, func_call) File "/usr/lib/python3.10/concurrent/futures/thread.py", line 58, in run result = self.fn(self.args, self.kwargs) File "/home/user/app/liveportrait/live_portrait_pipeline.py", line 38, in init self.live_portrait_wrapper: LivePortraitWrapper = LivePortraitWrapper(cfg=inference_cfg) File "/home/user/app/liveportrait/live_portrait_wrapper.py", line 28, in init self.appearance_feature_extractor = load_model(cfg.checkpoint_F, model_config, cfg.device_id, 'appearance_feature_extractor') File "/home/user/app/liveportrait/utils/helper.py", line 97, in load_model model = AppearanceFeatureExtractor(model_params).cuda(device) File "/home/user/.local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 916, in cuda return self._apply(lambda t: t.cuda(device)) File "/home/user/.local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 780, in _apply module._apply(fn) File "/home/user/.local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 780, in _apply module._apply(fn) File "/home/user/.local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 805, in _apply param_applied = fn(param) File "/home/user/.local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 916, in
return self._apply(lambda t: t.cuda(device))
File "/home/user/.local/lib/python3.10/site-packages/torch/cuda/init.py", line 314, in _lazy_init
torch._C._cuda_init()
RuntimeError: Found no NVIDIA driver on your system. Please check that you have an NVIDIA GPU and installed a driver from http://www.nvidia.com/Download/index.aspx
2024-10-09 05:36:15,720 - main - CRITICAL - 🚨 FATAL: Failed to start the app: Found no NVIDIA driver on your system. Please check that you have an NVIDIA GPU and installed a driver from http://www.nvidia.com/Download/index.aspx
2024-10-09 05:36:15,720 - main - ERROR - Full traceback:
Traceback (most recent call last):
File "/home/user/app/app.py", line 140, in
app = asyncio.run(initialize_app())
File "/usr/lib/python3.10/asyncio/runners.py", line 44, in run
return loop.run_until_complete(main)
File "/usr/lib/python3.10/asyncio/base_events.py", line 649, in run_until_complete
return future.result()
File "/home/user/app/app.py", line 114, in initialize_app
live_portrait = await initialize_models()
File "/home/user/app/loader.py", line 157, in initialize_models
live_portrait = await loader.load_live_portrait()
File "/home/user/app/loader.py", line 131, in load_live_portrait
live_portrait_pipeline = await asyncio.to_thread(
File "/usr/lib/python3.10/asyncio/threads.py", line 25, in to_thread
return await loop.run_in_executor(None, func_call)
File "/usr/lib/python3.10/concurrent/futures/thread.py", line 58, in run
result = self.fn( self.args, self.kwargs)
File "/home/user/app/liveportrait/live_portrait_pipeline.py", line 38, in init
self.live_portrait_wrapper: LivePortraitWrapper = LivePortraitWrapper(cfg=inference_cfg)
File "/home/user/app/liveportrait/live_portrait_wrapper.py", line 28, in init
self.appearance_feature_extractor = load_model(cfg.checkpoint_F, model_config, cfg.device_id, 'appearance_feature_extractor')
File "/home/user/app/liveportrait/utils/helper.py", line 97, in load_model
model = AppearanceFeatureExtractor(model_params).cuda(device)
File "/home/user/.local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 916, in cuda
return self._apply(lambda t: t.cuda(device))
File "/home/user/.local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 780, in _apply
module._apply(fn)
File "/home/user/.local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 780, in _apply
module._apply(fn)
File "/home/user/.local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 805, in _apply
param_applied = fn(param)
File "/home/user/.local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 916, in
return self._apply(lambda t: t.cuda(device))
File "/home/user/.local/lib/python3.10/site-packages/torch/cuda/init.py", line 314, in _lazy_init
torch._C._cuda_init()
RuntimeError: Found no NVIDIA driver on your system. Please check that you have an NVIDIA GPU and installed a driver from http://www.nvidia.com/Download/index.aspx
2024-10-09 05:36:16,072 - httpx - INFO - HTTP Request: GET https://api.gradio.app/gradio-messaging/en "HTTP/1.1 200 OK"