Skip to content

Commit

Permalink
Downgrade to CuDNN 8.9.0
Browse files Browse the repository at this point in the history
  • Loading branch information
gs-olive committed Jun 15, 2023
1 parent f0f1df3 commit 592f97a
Show file tree
Hide file tree
Showing 3 changed files with 9 additions and 9 deletions.
10 changes: 5 additions & 5 deletions .circleci/config.yml
Original file line number Diff line number Diff line change
Expand Up @@ -94,7 +94,7 @@ commands:
- << parameters.install-dir >>

install-cudnn:
description: "Install CUDNN 8.9.1"
description: "Install CUDNN 8.9.0"
parameters:
os:
type: string
Expand All @@ -104,7 +104,7 @@ commands:
default: "x86_64"
cudnn-version:
type: string
default: "8.9.1.23"
default: "8.9.0.131"
cuda-version:
type: string
default: "cuda12.1"
Expand Down Expand Up @@ -183,7 +183,7 @@ commands:
default: "cuda12.0"
cudnn-version:
type: string
default: "8.9.1.23"
default: "8.9.0.131"
trt-version-short:
type: string
default: "8.6.1"
Expand Down Expand Up @@ -235,7 +235,7 @@ commands:
default: "8.6.1"
cudnn-version-long:
type: string
default: "8.9.1.23"
default: "8.9.0.131"
steps:
- run:
name: Set up python environment
Expand Down Expand Up @@ -1356,7 +1356,7 @@ parameters:
default: true
cudnn-version:
type: string
default: "8.9.1.23"
default: "8.9.0.131"
trt-version-short:
type: string
default: "8.6.1"
Expand Down
2 changes: 1 addition & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -118,7 +118,7 @@ These are the following dependencies used to verify the testcases. Torch-TensorR
- Bazel 5.2.0
- Libtorch 2.1.0.dev20230613 (built with CUDA 12.1)
- CUDA 12.1
- cuDNN 8.9.1
- cuDNN 8.9.0
- TensorRT 8.6.1

## Prebuilt Binaries and Wheel files
Expand Down
6 changes: 3 additions & 3 deletions WORKSPACE
Original file line number Diff line number Diff line change
Expand Up @@ -71,10 +71,10 @@ http_archive(
http_archive(
name = "cudnn",
build_file = "@//third_party/cudnn/archive:BUILD",
sha256 = "35163c5c542be0c511738b27e25235193cbeedc5e0e006e44b1cdeaf1922e83e",
strip_prefix = "cudnn-linux-x86_64-8.9.1.23_cuda12-archive",
sha256 = "477631002be61022b60961cba0a501271507a93f81d6b08384bc320cb8706c98",
strip_prefix = "cudnn-linux-x86_64-8.9.0.131_cuda12-archive",
urls = [
"https://developer.nvidia.com/downloads/compute/cudnn/secure/8.9.1/local_installers/12.x/cudnn-linux-x86_64-8.9.1.23_cuda12-archive.tar.xz",
"https://developer.nvidia.com/downloads/compute/cudnn/secure/8.9.0/local_installers/12.x/cudnn-linux-x86_64-8.9.0.131_cuda12-archive.tar.xz",
],
)

Expand Down

0 comments on commit 592f97a

Please sign in to comment.