Can't initialize NVML error with RVC project

I’m trying to run this project on my local machine: GitHub - RVC-Project/Retrieval-based-Voice-Conversion-WebUI: Voice data <= 10 mins can also be used to train a good VC model!

I followed all the instructions on the read me of the project and when I try to run the project, I’m getting this error:

/home//.local/lib/python3.10/site-packages/torch/cuda/init.py:611: UserWarning: Can’t initialize NVML
warnings.warn(“Can’t initialize NVML”)
/home/
/.local/lib/python3.10/site-packages/torch/cuda/init.py:740: UserWarning: CUDA initialization: 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: Alternatively, go to: to install a PyTorch version that has been compiled with your version of the CUDA driver. (Triggered internally at …/c10/cuda/CUDAFunctions.cpp:108.)
return torch._C._cuda_getDeviceCount() if nvml_count < 0 else nvml_count
2023-12-23 16:12:24.342762: I tensorflow/core/platform/cpu_feature_guard.cc:182] This TensorFlow binary is optimized to use available CPU instructions in performance-critical operations.
To enable the following instructions: AVX2 FMA, in other operations, rebuild TensorFlow with the appropriate compiler flags.
2023-12-23 16:12:26 | INFO | configs.config | No supported Nvidia GPU found
overwrite preprocess and configs.json
2023-12-23 16:12:26 | INFO | configs.config | Use cpu instead
is_half:False, device:cpu
/home/*/.local/lib/python3.10/site-packages/matplotlib/projections/init.py:63: UserWarning: Unable to import Axes3D. This may be due to multiple versions of Matplotlib being installed (e.g. as a system package and as a pip package). As a result, the 3D projection is not available.
warnings.warn("Unable to import Axes3D. This may be due to multiple versions of "
2023-12-23 16:12:27 | INFO | main | Use Language: en_US
Running on local URL: http://0.0.0.0:7865

The read me section of the project asks me to run this: “pip install torch torchvision torchaudio --index-url 5.4.2”

But this link on pytorch is invalid.

I tried running the latest pytorch ROCm link and I get this error:

DEPRECATION: omegaconf 2.0.6 has a non-standard dependency specifier PyYAML>=5.1.*. pip 24.0 will enforce this behaviour change. A possible replacement is to upgrade to a newer version of omegaconf or contact the author to suggest that they release a version with a conforming dependency specifiers. Discussion can be found at github/pypa/pip/issues/12063

I doubt the can’t initialize NVML error is due to this.

This is my system:
OS: Ubuntu 22.06
CPU: Ryzen 3 3200G
RAM: 16GB
GPU: Radeon RX 6500 XT

How can I resolve this?

NVML is the NVIDIA Management Library and is used on NVIDIA GPUs. You might have installed the PyTorch binaries with CUDA support by mistake and might want to install the ones with rocm for AMD GPUs.

Thanks for this insight.

But running this Pytorch ROCm installation command is giving me errors

pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm5.7

This error: DEPRECATION: omegaconf 2.0.6 has a non-standard dependency specifier PyYAML>=5.1.*. pip 24.0 will enforce this behaviour change. A possible replacement is to upgrade to a newer version of omegaconf or contact the author to suggest that they release a version with a conforming dependency specifiers. Discussion can be found at github/pypa/pip/issues/12063

Should I uninstall the NVIDIA one?