Skip to content

Commit

Permalink
Update cuDNN installation instruction in README (pytorch#1409)
Browse files Browse the repository at this point in the history
Summary:
Pull Request resolved: pytorch#1409

As titled

Reviewed By: shintaro-iwasaki

Differential Revision: D40595839

fbshipit-source-id: e4992fb59699830e4b06107e585004a1cb0959ff
  • Loading branch information
sryap authored and facebook-github-bot committed Oct 21, 2022
1 parent 7429d93 commit 3c7f87e
Showing 1 changed file with 8 additions and 0 deletions.
8 changes: 8 additions & 0 deletions fbgemm_gpu/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -59,6 +59,7 @@ pip install fbgemm-gpu-nightly-cpu
## Build from source

Additional dependencies: currently cuDNN is required to be installed.
Please [download][4] and follow instructions [here][5] to install cuDNN.

```
# Requires PyTorch 1.13 or later
Expand All @@ -74,6 +75,11 @@ git submodule update --init --recursive
export CUDA_BIN_PATH=/usr/local/cuda-11.3/
export CUDACXX=/usr/local/cuda-11.3/bin/nvcc
# Specify cuDNN library and header paths. We tested CUDA 11.6 and 11.7 with
# cuDNN version 8.5.0.96
export CUDNN_LIBRARY=${HOME}/cudnn-linux-x86_64-8.5.0.96_cuda11-archive/lib
export CUDNN_INCLUDE_DIR=${HOME}/cudnn-linux-x86_64-8.5.0.96_cuda11-archive/include
# if using CUDA 10 or earliers set the location to the CUB installation directory
export CUB_DIR=${CUB_DIR}
# in fbgemm_gpu folder
Expand Down Expand Up @@ -181,3 +187,5 @@ FBGEMM is BSD licensed, as found in the [`LICENSE`](../LICENSE) file.
[1]:https://github.com/NVIDIA/cub
[2]:https://github.com/pytorch/pytorch
[3]:https://jinja.palletsprojects.com/en/2.11.x/
[4]:https://docs.nvidia.com/deeplearning/cudnn/install-guide/index.html#download
[5]:https://docs.nvidia.com/deeplearning/cudnn/install-guide/index.html#installlinux-tar

0 comments on commit 3c7f87e

Please sign in to comment.