Skip to content

Commit

Permalink
Fix cudnn installation (#1212)
Browse files Browse the repository at this point in the history
  • Loading branch information
csukuangfj authored Jun 16, 2023
1 parent 1a76309 commit 7ff6d89
Showing 1 changed file with 6 additions and 9 deletions.
15 changes: 6 additions & 9 deletions docs/source/installation/cuda-cudnn.rst
Original file line number Diff line number Diff line change
Expand Up @@ -425,10 +425,9 @@ Now, install ``cuDNN`` for CUDA 11.6.

.. code-block:: bash
wget https://huggingface.co/csukuangfj/cudnn/resolve/main/cudnn-11.3-linux-x64-v8.2.0.53.tgz
wget https://huggingface.co/csukuangfj/cudnn/resolve/main/cudnn-linux-x86_64-8.9.1.23_cuda11-archive.tar.xz
# Note: cudnn-11.3 works for CUDA 11.x
tar xvf cudnn-11.3-linux-x64-v8.2.0.53.tgz --strip-components=1 -C /ceph-sh1/fangjun/software/cuda-11.6.1
tar xvf cudnn-linux-x86_64-8.9.1.23_cuda11-archive.tar.xz --strip-components=1 -C /ceph-sh1/fangjun/software/cuda-11.6.1
Set environment variables for CUDA 11.6
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
Expand Down Expand Up @@ -499,10 +498,9 @@ Now, install ``cuDNN`` for CUDA 11.7.

.. code-block:: bash
wget https://huggingface.co/csukuangfj/cudnn/resolve/main/cudnn-11.3-linux-x64-v8.2.0.53.tgz
wget https://huggingface.co/csukuangfj/cudnn/resolve/main/cudnn-linux-x86_64-8.9.1.23_cuda11-archive.tar.xz
# Note: cudnn-11.3 works for CUDA 11.x
tar xvf cudnn-11.3-linux-x64-v8.2.0.53.tgz --strip-components=1 -C /ceph-sh1/fangjun/software/cuda-11.7.1
tar xvf cudnn-linux-x86_64-8.9.1.23_cuda11-archive.tar.xz --strip-components=1 -C /ceph-sh1/fangjun/software/cuda-11.7.1
Set environment variables for CUDA 11.7
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
Expand Down Expand Up @@ -573,10 +571,9 @@ Now, install ``cuDNN`` for CUDA 11.8.

.. code-block:: bash
wget https://huggingface.co/csukuangfj/cudnn/resolve/main/cudnn-11.3-linux-x64-v8.2.0.53.tgz
wget https://huggingface.co/csukuangfj/cudnn/resolve/main/cudnn-linux-x86_64-8.9.1.23_cuda11-archive.tar.xz
# Note: cudnn-11.3 works for CUDA 11.x
tar xvf cudnn-11.3-linux-x64-v8.2.0.53.tgz --strip-components=1 -C /star-fj/fangjun/software/cuda-11.8.0
tar xvf cudnn-linux-x86_64-8.9.1.23_cuda11-archive.tar.xz --strip-components=1 -C /star-fj/fangjun/software/cuda-11.8.0
Set environment variables for CUDA 11.8
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
Expand Down

0 comments on commit 7ff6d89

Please sign in to comment.