Skip to content

Commit

Permalink
Update INSTALL.rst (#915)
Browse files Browse the repository at this point in the history
* Update INSTALL.rst

Setting a few additional env variables to enable compilation from source *with CUDA GPU computation support enabled*
  • Loading branch information
alexei-v-ivanov authored Feb 8, 2022
1 parent cf32e2d commit 9e7b2a9
Showing 1 changed file with 12 additions and 0 deletions.
12 changes: 12 additions & 0 deletions INSTALL.rst
Original file line number Diff line number Diff line change
Expand Up @@ -44,6 +44,18 @@ From source
git clone https://github.com/k2-fsa/k2.git
cd k2
python3 setup.py install
From source (with CUDA support)
=========================

.. code-block:: bash
git clone https://github.com/k2-fsa/k2.git
cd k2
export K2_CMAKE_ARGS="-DK2_WITH_CUDA=ON -DCMAKE_BUILD_TYPE=Release"
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:/usr/local/cuda/lib:$LD_LIBRARY_PATH
export PATH=$PATH:/usr/local/cuda/bin
python3 setup.py install
Read `<https://k2.readthedocs.io/en/latest/installation/from_source.html>`_
to learn more
Expand Down

0 comments on commit 9e7b2a9

Please sign in to comment.