compatible with pytorch and cuda 12.4

[collyyang520]

Hello everyone:
I am Yang Ren shu. I recently encountered a version adjustment problem. Specifically, I encountered this problem
RuntimeError: NVML_SUCCESS == DriverAPI::get()->nvmlInit_v2_() INTERNAL ASSERT FAILED at "../c10/cuda/CUDACachingAllocator.cpp":963, please report a bug to PyTorch.
I have found related articles. It indicates that sudo reboot is required, but I am worried that someone is using or needs this version. I would like to ask how you adjusted it. Thank you

I found that pytorch only supports cuda up to 12.4, but I originally installed 12.6. I want to change it directly to

wget https://developer.download.nvidia.com/compute/cuda/12.4.0/local_installers/cuda_12.4.0_550.54.14_linux.run
sudo sh cuda_12.4.0_550.54.14_linux.run,

but I am worried about affecting other people, and the original conda install pytorch torchvision torchaudio cudatoolkit=12.4 -c pytorch is installed locally, but it cannot be installed successfully.

My worry is that I am in workstation, and I am in a virtual environment. I am fear that my change cuda will make other unhappy. I only want to run the pytorch with GPU to accelate.
By the way, I also a question:

/home/nthuuser/miniconda3/envs/earthquake/lib/python3.12/site-packages/torch/cuda/__init__.py:716: UserWarning: Can't initialize NVML
  warnings.warn("Can't initialize NVML")

Could someone help me?　　These two problem really make me frustrate.
Thanks in advanced.