NVIDIA GPU as dependency?

[Jumaku]

Hi there,

just tried to get PIXIE working. I tried to do so on a arch linux based system with the follwing GPU:

00:02.0 VGA compatible controller: Intel Corporation CometLake-U GT2 [UHD Graphics] (rev 02) Subsystem: Lenovo CometLake-U GT2 [UHD Graphics] Kernel driver in use: i915

and as it seems a NVIDIA GPU is required as shown in the output of the demo_animate_body.py script:

CUDA is not available! use CPU instead
total 1 images
total 6 images
Traceback (most recent call last):
  File "demos/demo_animate_body.py", line 150, in <module>
    main(parser.parse_args())
  File "demos/demo_animate_body.py", line 37, in main
    pixie = PIXIE(config = pixie_cfg, device=device)
  File "/home/juk/code/python/PIXIE/pixielib/pixie.py", line 50, in __init__
    self._create_model()
  File "/home/juk/code/python/PIXIE/pixielib/pixie.py", line 73, in _create_model
    self.Encoder[key] = HRNEncoder().to(self.device)
  File "/home/juk/.conda/envs/pixie-env/lib/python3.7/site-packages/torch/nn/modules
/module.py", line 607, in to
    return self._apply(convert)
  File "/home/juk/.conda/envs/pixie-env/lib/python3.7/site-packages/torch/nn/modules
/module.py", line 354, in _apply
    module._apply(fn)
  File "/home/juk/.conda/envs/pixie-env/lib/python3.7/site-packages/torch/nn/modules
/module.py", line 354, in _apply
    module._apply(fn)
  File "/home/juk/.conda/envs/pixie-env/lib/python3.7/site-packages/torch/nn/modules
/module.py", line 376, in _apply
    param_applied = fn(param)
  File "/home/juk/.conda/envs/pixie-env/lib/python3.7/site-packages/torch/nn/modules
/module.py", line 605, in convert
    return t.to(device, dtype if t.is_floating_point() else None, non_blocking)
  File "/home/juk/.conda/envs/pixie-env/lib/python3.7/site-packages/torch/cuda/__ini
t__.py", line 186, in _lazy_init
    _check_driver()
  File "/home/juk/.conda/envs/pixie-env/lib/python3.7/site-packages/torch/cuda/__ini
t__.py", line 68, in _check_driver
    http://www.nvidia.com/Download/index.aspx""")
AssertionError:
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

I guess mentioning that somewhere could some people save some time. If I am completely wrong and there is a way to get it running pls let me know :)

Here the content of my conda environment:

_libgcc_mutex             0.1                        main
_openmp_mutex             5.1                       1_gnu
ca-certificates           2023.05.30           h06a4308_0
certifi                   2022.12.7        py37h06a4308_0
chumpy                    0.70                     pypi_0    pypi
cycler                    0.11.0                   pypi_0    pypi
face-alignment            1.4.0                    pypi_0    pypi
fonttools                 4.38.0                   pypi_0    pypi
future                    0.18.3                   pypi_0    pypi
imageio                   2.31.1                   pypi_0    pypi
importlib-metadata        6.7.0                    pypi_0    pypi
kiwisolver                1.4.4                    pypi_0    pypi
kornia                    0.4.0                    pypi_0    pypi
libedit                   3.1.20221030         h5eee18b_0
libffi                    3.2.1             hf484d3e_1007
libgcc-ng                 11.2.0               h1234567_1
libgomp                   11.2.0               h1234567_1
libstdcxx-ng              11.2.0               h1234567_1
llvmlite                  0.39.1                   pypi_0    pypi
matplotlib                3.5.3                    pypi_0    pypi
ncurses                   6.4                  h6a678d5_0
networkx                  2.6.3                    pypi_0    pypi
ninja                     1.11.1                   pypi_0    pypi
numba                     0.56.4                   pypi_0    pypi
numpy                     1.21.6                   pypi_0    pypi
opencv-python             4.8.0.74                 pypi_0    pypi
openssl                   1.1.1v               h7f8727e_0
packaging                 23.1                     pypi_0    pypi
pillow                    9.5.0                    pypi_0    pypi
pip                       22.3.1           py37h06a4308_0
pyparsing                 3.1.1                    pypi_0    pypi
python                    3.7.4                h265db76_1
python-dateutil           2.8.2                    pypi_0    pypi
pywavelets                1.3.0                    pypi_0    pypi
pyyaml                    5.1.1                    pypi_0    pypi
readline                  7.0                  h7b6447c_5
scikit-image              0.19.3                   pypi_0    pypi
scipy                     1.7.3                    pypi_0    pypi
setuptools                65.6.3           py37h06a4308_0
six                       1.16.0                   pypi_0    pypi
sqlite                    3.33.0               h62c20be_0
tifffile                  2021.11.2                pypi_0    pypi
tk                        8.6.12               h1ccaba5_0
torch                     1.6.0                    pypi_0    pypi
torchaudio                0.6.0                    pypi_0    pypi
torchvision               0.7.0                    pypi_0    pypi
tqdm                      4.65.1                   pypi_0    pypi
typing-extensions         4.7.1                    pypi_0    pypi
wheel                     0.38.4           py37h06a4308_0
xz                        5.4.2                h5eee18b_0
yacs                      0.1.8                    pypi_0    pypi
zipp                      3.15.0                   pypi_0    pypi
zlib                      1.2.13               h5eee18b_0