Skip to content

Commit

Permalink
Update README to contain instructions on how to install mkldnn for Li…
Browse files Browse the repository at this point in the history
…nux (pytorch#7625)
  • Loading branch information
cpuhrsch authored and ezyang committed May 16, 2018
1 parent 3c9ded0 commit 330a725
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -175,6 +175,7 @@ export CMAKE_PREFIX_PATH="$(dirname $(which conda))/../" # [anaconda root direct

# Install basic dependencies
conda install numpy pyyaml mkl mkl-include setuptools cmake cffi typing
conda install -c intel mkl-dnn

# Add LAPACK support for the GPU
conda install -c pytorch magma-cuda80 # or magma-cuda90 if CUDA 9
Expand Down

0 comments on commit 330a725

Please sign in to comment.