search

farewellthree / PPLLaVA

Official GPU implementation of the paper "PPLLaVA: Varied Video Sequence Understanding With Prompt Guidance"
Apache License 2.0
84 stars 3 forks source link

CUDA Version #1

Closed zhong-yi-lin closed 3 days ago

zhong-yi-lin commented 1 week ago

I am trying to run the scripts with CUDA 11.7 installed, but I keep getting the error : RuntimeError: The NVIDIA driver on your system is too old (found version 11070). Please update your GPU driver by downloading and installing a new version from the URL: http://www.nvidia.com/Download/index.aspx Alternatively, go to: https://pytorch.org/ to install a PyTorch version that has been compiled with your version of the CUDA driver.

So, what's the right version? Thank you.

farewellthree commented 6 days ago

Hi, torch>=2.0 requires CUDA>=11.8, and the newest qwen model seems requiring torch>=2.1.2