message”: “NVML_SUCCESS == DriverAPI::get()->nvmlInit_v2_()

“message”: “NVML_SUCCESS == DriverAPI::get()->nvmlInit_v2_() INTERNAL ASSERT FAILED at \”../c10/cuda/CUDACachingAllocator.cpp\”:799, please report a bug to PyTorch. “,

Solution:

Restart the computer. Hope it works else run this in the conda environment

python -m torch.utils.collect_env