Skip to content

Commit d5bd60e

Browse files
committed
Test upgrade to CuDNN 8.8
1 parent 8f1c7c4 commit d5bd60e

File tree

2 files changed

+9
-8
lines changed

2 files changed

+9
-8
lines changed

.circleci/config.yml

Lines changed: 6 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -112,10 +112,10 @@ commands:
112112
default: "x86_64"
113113
cudnn-version:
114114
type: string
115-
default: "8.5.0.96"
115+
default: "8.8.0.121"
116116
cuda-version:
117117
type: string
118-
default: "cuda11.7"
118+
default: "cuda11.8"
119119
steps:
120120
- run:
121121
name: Install CUDNN
@@ -200,7 +200,7 @@ commands:
200200
default: "cuda11.8"
201201
cudnn-version:
202202
type: string
203-
default: "8.5.0.96"
203+
default: "8.8.0.121"
204204
trt-version-short:
205205
type: string
206206
default: "8.6.0"
@@ -252,7 +252,7 @@ commands:
252252
default: "8.6.0"
253253
cudnn-version-long:
254254
type: string
255-
default: "8.5.0.96"
255+
default: "8.8.0.121"
256256
steps:
257257
- run:
258258
name: Set up python environment
@@ -474,6 +474,7 @@ commands:
474474
- run: mkdir -p /tmp/artifacts
475475
- run:
476476
name: Run core / C++ tests
477+
no_output_timeout: 15m
477478
environment:
478479
LD_LIBRARY_PATH: "/opt/circleci/.pyenv/versions/3.9.4/lib/python3.9/site-packages/torch_tensorrt.libs:/home/circleci/project/bazel-project/external/libtorch_pre_cxx11_abi/lib/:/home/circleci/project/bazel-project/external/tensorrt/lib/:/usr/local/cuda-11.8/lib64/:$LD_LIBRARY_PATH"
479480
command: |
@@ -1217,7 +1218,7 @@ parameters:
12171218
default: "https://download.pytorch.org/whl/cu117"
12181219
cudnn-version:
12191220
type: string
1220-
default: "8.5.0.96"
1221+
default: "8.8.0.121"
12211222
trt-version-short:
12221223
type: string
12231224
default: "8.6.0"

WORKSPACE

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -71,10 +71,10 @@ http_archive(
7171
http_archive(
7272
name = "cudnn",
7373
build_file = "@//third_party/cudnn/archive:BUILD",
74-
sha256 = "5454a6fd94f008728caae9adad993c4e85ef36302e26bce43bea7d458a5e7b6d",
75-
strip_prefix = "cudnn-linux-x86_64-8.5.0.96_cuda11-archive",
74+
sha256 = "36fff137153ef73e6ee10bfb07f4381240a86fb9fb78ce372414b528cbab2293",
75+
strip_prefix = "cudnn-linux-x86_64-8.8.0.121_cuda11-archive",
7676
urls = [
77-
"https://developer.nvidia.com/compute/cudnn/secure/8.5.0/local_installers/11.7/cudnn-linux-x86_64-8.5.0.96_cuda11-archive.tar.xz",
77+
"https://developer.download.nvidia.com/compute/cudnn/secure/8.8.0/local_installers/11.8/cudnn-linux-x86_64-8.8.0.121_cuda11-archive.tar.xz",
7878
],
7979
)
8080

0 commit comments

Comments
 (0)