Skip to content

Commit 9e7b2a9

Browse files
Update INSTALL.rst (#915)
* Update INSTALL.rst Setting a few additional env variables to enable compilation from source *with CUDA GPU computation support enabled*
1 parent cf32e2d commit 9e7b2a9

File tree

1 file changed

+12
-0
lines changed

1 file changed

+12
-0
lines changed

INSTALL.rst

+12
Original file line numberDiff line numberDiff line change
@@ -44,6 +44,18 @@ From source
4444
git clone https://github.com/k2-fsa/k2.git
4545
cd k2
4646
python3 setup.py install
47+
48+
From source (with CUDA support)
49+
=========================
50+
51+
.. code-block:: bash
52+
53+
git clone https://github.com/k2-fsa/k2.git
54+
cd k2
55+
export K2_CMAKE_ARGS="-DK2_WITH_CUDA=ON -DCMAKE_BUILD_TYPE=Release"
56+
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:/usr/local/cuda/lib:$LD_LIBRARY_PATH
57+
export PATH=$PATH:/usr/local/cuda/bin
58+
python3 setup.py install
4759
4860
Read `<https://k2.readthedocs.io/en/latest/installation/from_source.html>`_
4961
to learn more

0 commit comments

Comments
 (0)