Skip to content

Commit

Permalink
Add CUDA_VISIBLE_DEVICES warning explanation so it can be avoided
Browse files Browse the repository at this point in the history
  • Loading branch information
gonidelis authored Nov 18, 2024
1 parent e8ac56a commit aba9ac4
Showing 1 changed file with 4 additions and 0 deletions.
4 changes: 4 additions & 0 deletions docs/cub/benchmarking.rst
Original file line number Diff line number Diff line change
Expand Up @@ -174,6 +174,10 @@ Therefore, it's critical that you run it in a clean build directory without any
Running cmake is enough. Alternatively, you can also clean your build directory with.
Furthermore, the tuning scripts require some additional python dependencies, which you have to install.

To select the appropriate CUDA GPU, first identify the GPU ID by running `nvidia-smi`, then set the
desired GPU using `export CUDA_VISIBLE_DEVICES=x`, where `x` is the ID of the GPU you want to use (e.g., `1`).
This ensures your application uses only the specified GPU.

.. code-block:: bash
ninja clean
Expand Down

0 comments on commit aba9ac4

Please sign in to comment.