|
25 | 25 |
|
26 | 26 |
|
27 | 27 | <link rel="icon" href="../../../assets/ms_icon.png"> |
28 | | - <meta name="generator" content="mkdocs-1.6.1, mkdocs-material-9.7.1"> |
| 28 | + <meta name="generator" content="mkdocs-1.6.1, mkdocs-material-9.7.5"> |
29 | 29 |
|
30 | 30 |
|
31 | 31 |
|
|
1796 | 1796 | </label> |
1797 | 1797 | <ul class="md-nav__list" data-md-component="toc" data-md-scrollfix> |
1798 | 1798 |
|
| 1799 | + <li class="md-nav__item"> |
| 1800 | + <a href="#environment-variable-override" class="md-nav__link"> |
| 1801 | + <span class="md-ellipsis"> |
| 1802 | + |
| 1803 | + Environment Variable Override |
| 1804 | + |
| 1805 | + </span> |
| 1806 | + </a> |
| 1807 | + |
| 1808 | +</li> |
| 1809 | + |
1799 | 1810 | <li class="md-nav__item"> |
1800 | 1811 | <a href="#prerequisites" class="md-nav__link"> |
1801 | 1812 | <span class="md-ellipsis"> |
|
4529 | 4540 | </label> |
4530 | 4541 | <ul class="md-nav__list" data-md-component="toc" data-md-scrollfix> |
4531 | 4542 |
|
| 4543 | + <li class="md-nav__item"> |
| 4544 | + <a href="#environment-variable-override" class="md-nav__link"> |
| 4545 | + <span class="md-ellipsis"> |
| 4546 | + |
| 4547 | + Environment Variable Override |
| 4548 | + |
| 4549 | + </span> |
| 4550 | + </a> |
| 4551 | + |
| 4552 | +</li> |
| 4553 | + |
4532 | 4554 | <li class="md-nav__item"> |
4533 | 4555 | <a href="#prerequisites" class="md-nav__link"> |
4534 | 4556 | <span class="md-ellipsis"> |
|
4673 | 4695 |
|
4674 | 4696 | <h1 id="gpu-acceleration">GPU Acceleration</h1> |
4675 | 4697 | <p>Presidio supports GPU acceleration for NLP models, which can significantly improve performance when processing large volumes of text. GPU is automatically detected and used when available—no code changes required.</p> |
| 4698 | +<h2 id="environment-variable-override">Environment Variable Override</h2> |
| 4699 | +<p>You can explicitly control device selection using the <code>PRESIDIO_DEVICE</code> environment variable. Any valid PyTorch device string is accepted:</p> |
| 4700 | +<div class="highlight"><pre><span></span><code><span class="c1"># Force CPU usage</span> |
| 4701 | +<span class="nb">export</span><span class="w"> </span><span class="nv">PRESIDIO_DEVICE</span><span class="o">=</span>cpu |
| 4702 | + |
| 4703 | +<span class="c1"># Force GPU usage</span> |
| 4704 | +<span class="nb">export</span><span class="w"> </span><span class="nv">PRESIDIO_DEVICE</span><span class="o">=</span>cuda |
| 4705 | + |
| 4706 | +<span class="c1"># Use a specific GPU</span> |
| 4707 | +<span class="nb">export</span><span class="w"> </span><span class="nv">PRESIDIO_DEVICE</span><span class="o">=</span>cuda:0 |
| 4708 | +<span class="nb">export</span><span class="w"> </span><span class="nv">PRESIDIO_DEVICE</span><span class="o">=</span>cuda:1 |
| 4709 | + |
| 4710 | +<span class="c1"># Or when running Docker containers</span> |
| 4711 | +docker<span class="w"> </span>run<span class="w"> </span>-e<span class="w"> </span><span class="nv">PRESIDIO_DEVICE</span><span class="o">=</span>cuda<span class="w"> </span>mcr.microsoft.com/presidio-analyzer:latest |
| 4712 | +</code></pre></div> |
| 4713 | +<div class="admonition note"> |
| 4714 | +<p class="admonition-title">Priority</p> |
| 4715 | +<p>When <code>PRESIDIO_DEVICE</code> is set, it takes priority over automatic detection. If not set (or empty), Presidio will auto-detect available hardware.</p> |
| 4716 | +</div> |
4676 | 4717 | <h2 id="prerequisites">Prerequisites</h2> |
4677 | 4718 | <h3 id="hardware-requirements">Hardware Requirements</h3> |
4678 | 4719 | <table> |
|
0 commit comments