123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237 |
- <!DOCTYPE html>
- <html lang="en">
- <head>
- <meta charset="utf-8" />
- <meta name="viewport" content="width=device-width, initial-scale=1.0" /><meta name="generator" content="Docutils 0.17.1: http://docutils.sourceforge.net/" />
- <title>draugr.torch_utilities.system.device — Draugr 1.0.1 documentation</title>
- <link rel="stylesheet" type="text/css" href="../_static/pygments.css" />
- <link rel="stylesheet" type="text/css" href="../_static/alabaster.css" />
- <link rel="stylesheet" type="text/css" href="../_static/graphviz.css" />
- <script data-url_root="../" id="documentation_options" src="../_static/documentation_options.js"></script>
- <script src="../_static/jquery.js"></script>
- <script src="../_static/underscore.js"></script>
- <script src="../_static/_sphinx_javascript_frameworks_compat.js"></script>
- <script src="../_static/doctools.js"></script>
- <link rel="canonical" href="pything.github.io/draugr/generated/draugr.torch_utilities.system.device.html" />
- <link rel="index" title="Index" href="../genindex.html" />
- <link rel="search" title="Search" href="../search.html" />
- <link rel="next" title="draugr.torch_utilities.system.device.global_torch_device" href="draugr.torch_utilities.system.device.global_torch_device.html" />
- <link rel="prev" title="draugr.torch_utilities.system.data_type.set_global_torch_dtype" href="draugr.torch_utilities.system.data_type.set_global_torch_dtype.html" />
-
- <link rel="stylesheet" href="../_static/custom.css" type="text/css" />
-
-
- <meta name="viewport" content="width=device-width, initial-scale=0.9, maximum-scale=0.9" />
- </head><body>
-
- <div class="document">
- <div class="documentwrapper">
- <div class="bodywrapper">
-
- <div class="body" role="main">
-
- <section id="module-draugr.torch_utilities.system.device">
- <span id="draugr-torch-utilities-system-device"></span><h1>draugr.torch_utilities.system.device<a class="headerlink" href="#module-draugr.torch_utilities.system.device" title="Permalink to this heading">¶</a></h1>
- <p>Created on 15/11/2019</p>
- <p class="rubric">Functions</p>
- <table class="autosummary longtable docutils align-default">
- <colgroup>
- <col style="width: 10%" />
- <col style="width: 90%" />
- </colgroup>
- <tbody>
- <tr class="row-odd"><td><p><a class="reference internal" href="draugr.torch_utilities.system.device.global_torch_device.html#draugr.torch_utilities.system.device.global_torch_device" title="draugr.torch_utilities.system.device.global_torch_device"><code class="xref py py-obj docutils literal notranslate"><span class="pre">global_torch_device</span></code></a>([device_preference, ...])</p></td>
- <td><p>first time call stores to device for global reference, later call must explicitly manually override!</p></td>
- </tr>
- <tr class="row-even"><td><p><a class="reference internal" href="draugr.torch_utilities.system.device.select_cuda_device.html#draugr.torch_utilities.system.device.select_cuda_device" title="draugr.torch_utilities.system.device.select_cuda_device"><code class="xref py py-obj docutils literal notranslate"><span class="pre">select_cuda_device</span></code></a>(cuda_device_idx)</p></td>
- <td><p><dl class="field-list simple">
- <dt class="field-odd">param cuda_device_idx</dt>
- <dd class="field-odd"><p></p></dd>
- </dl>
- </p></td>
- </tr>
- <tr class="row-odd"><td><p><a class="reference internal" href="draugr.torch_utilities.system.device.get_gpu_usage_mb.html#draugr.torch_utilities.system.device.get_gpu_usage_mb" title="draugr.torch_utilities.system.device.get_gpu_usage_mb"><code class="xref py py-obj docutils literal notranslate"><span class="pre">get_gpu_usage_mb</span></code></a>()</p></td>
- <td><p><dl class="field-list simple">
- <dt class="field-odd">return</dt>
- <dd class="field-odd"><p></p></dd>
- </dl>
- </p></td>
- </tr>
- <tr class="row-even"><td><p><a class="reference internal" href="draugr.torch_utilities.system.device.auto_select_available_cuda_device.html#draugr.torch_utilities.system.device.auto_select_available_cuda_device" title="draugr.torch_utilities.system.device.auto_select_available_cuda_device"><code class="xref py py-obj docutils literal notranslate"><span class="pre">auto_select_available_cuda_device</span></code></a>([...])</p></td>
- <td><p>Auto selects the device with highest compute capability and with the requested memory available</p></td>
- </tr>
- <tr class="row-odd"><td><p><a class="reference internal" href="draugr.torch_utilities.system.device.set_global_torch_device.html#draugr.torch_utilities.system.device.set_global_torch_device" title="draugr.torch_utilities.system.device.set_global_torch_device"><code class="xref py py-obj docutils literal notranslate"><span class="pre">set_global_torch_device</span></code></a>(device)</p></td>
- <td><p><dl class="field-list simple">
- <dt class="field-odd">param device</dt>
- <dd class="field-odd"><p></p></dd>
- </dl>
- </p></td>
- </tr>
- <tr class="row-even"><td><p><a class="reference internal" href="draugr.torch_utilities.system.device.torch_clean_up.html#draugr.torch_utilities.system.device.torch_clean_up" title="draugr.torch_utilities.system.device.torch_clean_up"><code class="xref py py-obj docutils literal notranslate"><span class="pre">torch_clean_up</span></code></a>()</p></td>
- <td><p><strong>Destroy cuda state by emptying cache and collecting IPC.</strong></p></td>
- </tr>
- </tbody>
- </table>
- <p class="rubric">Classes</p>
- <table class="autosummary longtable docutils align-default">
- <colgroup>
- <col style="width: 10%" />
- <col style="width: 90%" />
- </colgroup>
- <tbody>
- <tr class="row-odd"><td><p><a class="reference internal" href="draugr.torch_utilities.system.device.TorchDeviceEnum.html#draugr.torch_utilities.system.device.TorchDeviceEnum" title="draugr.torch_utilities.system.device.TorchDeviceEnum"><code class="xref py py-obj docutils literal notranslate"><span class="pre">TorchDeviceEnum</span></code></a>(value)</p></td>
- <td><p>An enumeration.</p></td>
- </tr>
- </tbody>
- </table>
- </section>
- </div>
-
- </div>
- </div>
- <div class="sphinxsidebar" role="navigation" aria-label="main navigation">
- <div class="sphinxsidebarwrapper">
- <p class="logo"><a href="../index.html">
- <img class="logo" src="../_static/valknut.svg" alt="Logo"/>
- </a></p>
- <h1 class="logo"><a href="../index.html">Draugr</a></h1>
- <h3>Navigation</h3>
- <ul class="current">
- <li class="toctree-l1 current"><a class="reference internal" href="draugr.html">draugr</a><ul class="current">
- <li class="toctree-l2"><a class="reference internal" href="draugr.dist_is_editable.html">draugr.dist_is_editable</a></li>
- <li class="toctree-l2"><a class="reference internal" href="draugr.get_version.html">draugr.get_version</a></li>
- <li class="toctree-l2"><a class="reference internal" href="draugr.dlib_utilities.html">draugr.dlib_utilities</a></li>
- <li class="toctree-l2"><a class="reference internal" href="draugr.drawers.html">draugr.drawers</a></li>
- <li class="toctree-l2"><a class="reference internal" href="draugr.entry_points.html">draugr.entry_points</a></li>
- <li class="toctree-l2"><a class="reference internal" href="draugr.extensions.html">draugr.extensions</a></li>
- <li class="toctree-l2"><a class="reference internal" href="draugr.ffmpeg_utilities.html">draugr.ffmpeg_utilities</a></li>
- <li class="toctree-l2"><a class="reference internal" href="draugr.jax_utilities.html">draugr.jax_utilities</a></li>
- <li class="toctree-l2"><a class="reference internal" href="draugr.metrics.html">draugr.metrics</a></li>
- <li class="toctree-l2"><a class="reference internal" href="draugr.multiprocessing_utilities.html">draugr.multiprocessing_utilities</a></li>
- <li class="toctree-l2"><a class="reference internal" href="draugr.numpy_utilities.html">draugr.numpy_utilities</a></li>
- <li class="toctree-l2"><a class="reference internal" href="draugr.opencv_utilities.html">draugr.opencv_utilities</a></li>
- <li class="toctree-l2"><a class="reference internal" href="draugr.os_utilities.html">draugr.os_utilities</a></li>
- <li class="toctree-l2"><a class="reference internal" href="draugr.pandas_utilities.html">draugr.pandas_utilities</a></li>
- <li class="toctree-l2"><a class="reference internal" href="draugr.pygame_utilities.html">draugr.pygame_utilities</a></li>
- <li class="toctree-l2"><a class="reference internal" href="draugr.python_utilities.html">draugr.python_utilities</a></li>
- <li class="toctree-l2"><a class="reference internal" href="draugr.random_utilities.html">draugr.random_utilities</a></li>
- <li class="toctree-l2"><a class="reference internal" href="draugr.scipy_utilities.html">draugr.scipy_utilities</a></li>
- <li class="toctree-l2"><a class="reference internal" href="draugr.stopping.html">draugr.stopping</a></li>
- <li class="toctree-l2"><a class="reference internal" href="draugr.tensorboard_utilities.html">draugr.tensorboard_utilities</a></li>
- <li class="toctree-l2"><a class="reference internal" href="draugr.threading_utilities.html">draugr.threading_utilities</a></li>
- <li class="toctree-l2 current"><a class="reference internal" href="draugr.torch_utilities.html">draugr.torch_utilities</a><ul class="current">
- <li class="toctree-l3"><a class="reference internal" href="draugr.torch_utilities.architectures.html">draugr.torch_utilities.architectures</a></li>
- <li class="toctree-l3"><a class="reference internal" href="draugr.torch_utilities.datasets.html">draugr.torch_utilities.datasets</a></li>
- <li class="toctree-l3"><a class="reference internal" href="draugr.torch_utilities.distributions.html">draugr.torch_utilities.distributions</a></li>
- <li class="toctree-l3"><a class="reference internal" href="draugr.torch_utilities.evaluation.html">draugr.torch_utilities.evaluation</a></li>
- <li class="toctree-l3"><a class="reference internal" href="draugr.torch_utilities.exporting.html">draugr.torch_utilities.exporting</a></li>
- <li class="toctree-l3"><a class="reference internal" href="draugr.torch_utilities.generators.html">draugr.torch_utilities.generators</a></li>
- <li class="toctree-l3"><a class="reference internal" href="draugr.torch_utilities.images.html">draugr.torch_utilities.images</a></li>
- <li class="toctree-l3"><a class="reference internal" href="draugr.torch_utilities.opencv.html">draugr.torch_utilities.opencv</a></li>
- <li class="toctree-l3"><a class="reference internal" href="draugr.torch_utilities.operations.html">draugr.torch_utilities.operations</a></li>
- <li class="toctree-l3"><a class="reference internal" href="draugr.torch_utilities.optimisation.html">draugr.torch_utilities.optimisation</a></li>
- <li class="toctree-l3"><a class="reference internal" href="draugr.torch_utilities.persistence.html">draugr.torch_utilities.persistence</a></li>
- <li class="toctree-l3"><a class="reference internal" href="draugr.torch_utilities.sessions.html">draugr.torch_utilities.sessions</a></li>
- <li class="toctree-l3 current"><a class="reference internal" href="draugr.torch_utilities.system.html">draugr.torch_utilities.system</a><ul class="current">
- <li class="toctree-l4"><a class="reference internal" href="draugr.torch_utilities.system.constants.html">draugr.torch_utilities.system.constants</a></li>
- <li class="toctree-l4"><a class="reference internal" href="draugr.torch_utilities.system.data_type.html">draugr.torch_utilities.system.data_type</a></li>
- <li class="toctree-l4 current"><a class="current reference internal" href="#">draugr.torch_utilities.system.device</a><ul>
- <li class="toctree-l5"><a class="reference internal" href="draugr.torch_utilities.system.device.global_torch_device.html">draugr.torch_utilities.system.device.global_torch_device</a></li>
- <li class="toctree-l5"><a class="reference internal" href="draugr.torch_utilities.system.device.select_cuda_device.html">draugr.torch_utilities.system.device.select_cuda_device</a></li>
- <li class="toctree-l5"><a class="reference internal" href="draugr.torch_utilities.system.device.get_gpu_usage_mb.html">draugr.torch_utilities.system.device.get_gpu_usage_mb</a></li>
- <li class="toctree-l5"><a class="reference internal" href="draugr.torch_utilities.system.device.auto_select_available_cuda_device.html">draugr.torch_utilities.system.device.auto_select_available_cuda_device</a></li>
- <li class="toctree-l5"><a class="reference internal" href="draugr.torch_utilities.system.device.set_global_torch_device.html">draugr.torch_utilities.system.device.set_global_torch_device</a></li>
- <li class="toctree-l5"><a class="reference internal" href="draugr.torch_utilities.system.device.torch_clean_up.html">draugr.torch_utilities.system.device.torch_clean_up</a></li>
- <li class="toctree-l5"><a class="reference internal" href="draugr.torch_utilities.system.device.TorchDeviceEnum.html">draugr.torch_utilities.system.device.TorchDeviceEnum</a></li>
- </ul>
- </li>
- <li class="toctree-l4"><a class="reference internal" href="draugr.torch_utilities.system.info.html">draugr.torch_utilities.system.info</a></li>
- <li class="toctree-l4"><a class="reference internal" href="draugr.torch_utilities.system.loading.html">draugr.torch_utilities.system.loading</a></li>
- <li class="toctree-l4"><a class="reference internal" href="draugr.torch_utilities.system.seeding.html">draugr.torch_utilities.system.seeding</a></li>
- </ul>
- </li>
- <li class="toctree-l3"><a class="reference internal" href="draugr.torch_utilities.tensors.html">draugr.torch_utilities.tensors</a></li>
- <li class="toctree-l3"><a class="reference internal" href="draugr.torch_utilities.writers.html">draugr.torch_utilities.writers</a></li>
- </ul>
- </li>
- <li class="toctree-l2"><a class="reference internal" href="draugr.tqdm_utilities.html">draugr.tqdm_utilities</a></li>
- <li class="toctree-l2"><a class="reference internal" href="draugr.visualisation.html">draugr.visualisation</a></li>
- <li class="toctree-l2"><a class="reference internal" href="draugr.writers.html">draugr.writers</a></li>
- </ul>
- </li>
- </ul>
- <p class="caption" role="heading"><span class="caption-text">Notes</span></p>
- <ul>
- <li class="toctree-l1"><a class="reference internal" href="../getting_started.html">Getting Started</a></li>
- </ul>
- <div class="relations">
- <h3>Related Topics</h3>
- <ul>
- <li><a href="../index.html">Documentation overview</a><ul>
- <li><a href="draugr.html">draugr</a><ul>
- <li><a href="draugr.torch_utilities.html">draugr.torch_utilities</a><ul>
- <li><a href="draugr.torch_utilities.system.html">draugr.torch_utilities.system</a><ul>
- <li>Previous: <a href="draugr.torch_utilities.system.data_type.set_global_torch_dtype.html" title="previous chapter">draugr.torch_utilities.system.data_type.set_global_torch_dtype</a></li>
- <li>Next: <a href="draugr.torch_utilities.system.device.global_torch_device.html" title="next chapter">draugr.torch_utilities.system.device.global_torch_device</a></li>
- </ul></li>
- </ul></li>
- </ul></li>
- </ul></li>
- </ul>
- </div>
- <div id="searchbox" style="display: none" role="search">
- <h3 id="searchlabel">Quick search</h3>
- <div class="searchformwrapper">
- <form class="search" action="../search.html" method="get">
- <input type="text" name="q" aria-labelledby="searchlabel" autocomplete="off" autocorrect="off" autocapitalize="off" spellcheck="false"/>
- <input type="submit" value="Go" />
- </form>
- </div>
- </div>
- <script>document.getElementById('searchbox').style.display = "block"</script>
- </div>
- </div>
- <div class="clearer"></div>
- </div>
- <div class="footer">
- ©.
-
- |
- Powered by <a href="http://sphinx-doc.org/">Sphinx 5.0.2</a>
- & <a href="https://github.com/bitprophet/alabaster">Alabaster 0.7.12</a>
-
- |
- <a href="../_sources/generated/draugr.torch_utilities.system.device.rst.txt"
- rel="nofollow">Page source</a>
- </div>
-
-
- </body>
- </html>
|