vakyansh-models
vakyansh-models copied to clipboard
CUDA dependency
I have installed the required dependencies with flag USE_CUDA=0. However, single file inference doesn't work for english_infer.pt with the error message
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
Any pointers to fix the issue would be great.
Try install PyTorch version with CUDA support. It will work.