Can run demo but training error can not find the package
lucy0427 opened this issue · comments
The required dependencies are installed, the demo can be trained normally, and the output is correct. But when I try to train, I get an error saying cupy can't be found (loguru can't be found either, comment it to show cupy can't be found)
Have you run "pip3 install cupy_cuda102==9.4.0" and "pip3 install -r requirements.txt"?
The required dependencies are installed, the demo can be trained normally, and the output is correct. But when I try to train, I get an error saying cupy can't be found (loguru can't be found either, comment it to show cupy can't be found)
Have you run "pip3 install cupy_cuda102==9.4.0" and "pip3 install -r requirements.txt"?
Yes! I ran and installed these dependencies, but for cupy I installed cupy_cuda121==12.3.0 because the server I was using had cuda version 12.1