python - 0 <= device && device <device_num INTERNAL ASSERT FAILED at

Question

• The computer was able to run the code on the GPU.

• I did a computer restart.

• The computer cannot run the code and is recording the following error.

---

RuntimeError Traceback (most recent call last)
<ipython-input-8-186b7088820a> in <module>
      6 torch.cuda.empty_cache ()
      7
----> 8 print (torch.cuda.memory_allocated (0))

~ / anaconda3 / lib / python3.7 / site-packages / torch / cuda / memory.py in memory_allocated (device)
    276 details about GPU memory management.
    277 "" "
-> 278 return memory_stats (device = device) ["allocated_bytes.all.current"]
    279
    280

~ / anaconda3 / lib / python3.7 / site-packages / torch / cuda / memory.py in memory_stats (device)
    157 result.append ((prefix, obj))
    158
-> 159 stats = memory_stats_as_nested_dict (device = device)
    160 _recurse_add_to_result ("", stats)
    161 result.sort ()

~ / anaconda3 / lib / python3.7 / site-packages / torch / cuda / memory.py in memory_stats_as_nested_dict (device)
    167 "Returns the result of: func:` ~ torch.cuda.memory_stats` as a nested dictionary. "" "
    168 device = _get_device_index (device, optional = True)
-> 169 return torch._C._cuda_memoryStats (device)
    170
    171

RuntimeError: 0 <= device && device <device_num INTERNAL ASSERT FAILED at /pytorch/c10/cuda/CUDACachingAllocator.cpp:879, please report a bug to PyTorch. Invalid device argument. 

question from:https://stackoverflow.com/questions/65617192/0-device-device-device-num-internal-assert-failed-at

Answer 1

Waitting for answers