Skip to content

Commit

Permalink
Update MKL instruction
Browse files Browse the repository at this point in the history
  • Loading branch information
amberfly authored Dec 7, 2017
1 parent 16ff1fb commit 1bf7e7b
Showing 1 changed file with 9 additions and 1 deletion.
10 changes: 9 additions & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -139,9 +139,17 @@ echo 'export LD_LIBRARY_PATH=/usr/local/cuda/extras/CUPTI/lib64:$LD_LIBRARY_PATH

### MKL

MKL is Intel's deep learning kernal library. If you don't have it, follow the [instructions here](https://github.com/01org/mkl-dnn) to install it. The `make install` step may require `sudo`. After installing, add `/usr/local/lib` (or your custom install path) to `LD_LIBRARY_PATH`:
MKL is [Intel's deep learning kernal library](https://github.com/01org/mkl-dnn), which makes training neural nets on CPU much faster. If you don't have it, install it like the following:

```sh
# If you don't have cmake
sudo apt install cmake

git clone https://github.com/01org/mkl-dnn.git
cd mkl-dnn/scripts && ./prepare_mkl.sh && cd ..
mkdir -p build && cd build && cmake .. && make
sudo make install

echo 'export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/lib' >> ~/.bashrc
```

Expand Down

0 comments on commit 1bf7e7b

Please sign in to comment.