Почетна Преглед Помоћ
Пријавите се
cnheider
/
draugr
огледало од https://github.com/cnheider/draugr
Прати 1
Волим 1
Креирај огранак 0
Датотеке Дискусије 0 Вики
Грана: master
Гране Ознаке
draugr
/
docs
/
source
/
generated
/
draugr.torch_utilities.system.device.rst

draugr.torch_utilities.system.device.rst 558 B
Пермалинк Историја Датотека

12345678910111213141516171819202122232425262728293031323334353637383940414243444546 
  1. draugr.torch\_utilities.system.device
    2. 

  2. =====================================
    3. 

  3. 

  4. 

  5. 

  6. .. automodule:: draugr.torch_utilities.system.device
    7. 

  7. 

  8.    
    9. 

  9.    
    10. 

  10.    
    11. 

  11. 

  12.    
    13. 

  13.    
    14. 

  14.    .. rubric:: Functions
    15. 

  15. 

  16.    .. autosummary::
    17. 

  17.       :toctree:
    18. 

  18.    
    19. 

  19.       global_torch_device
    20. 

  20.       select_cuda_device
    21. 

  21.       get_gpu_usage_mb
    22. 

  22.       auto_select_available_cuda_device
    23. 

  23.       set_global_torch_device
    24. 

  24.       torch_clean_up
    25. 

  25.    
    26. 

  26.    
    27. 

  27. 

  28.    
    29. 

  29.    
    30. 

  30.    .. rubric:: Classes
    31. 

  31. 

  32.    .. autosummary::
    33. 

  33.       :toctree:
    34. 

  34.       :template: custom_autosummary/class.rst
    35. 

  35.    
    36. 

  36.       TorchDeviceEnum
    37. 

  37.    
    38. 

  38.    
    39. 

  39. 

  40.    
    41. 

  41.    
    42. 

  42.    
    43. 

  43. 

  44. 

  45. 

  46.