File tree Expand file tree Collapse file tree 2 files changed +9
-8
lines changed Expand file tree Collapse file tree 2 files changed +9
-8
lines changed Original file line number Diff line number Diff line change @@ -112,10 +112,10 @@ commands:
112
112
default : " x86_64"
113
113
cudnn-version :
114
114
type : string
115
- default : " 8.5 .0.96 "
115
+ default : " 8.8 .0.121 "
116
116
cuda-version :
117
117
type : string
118
- default : " cuda11.7 "
118
+ default : " cuda11.8 "
119
119
steps :
120
120
- run :
121
121
name : Install CUDNN
@@ -200,7 +200,7 @@ commands:
200
200
default : " cuda11.8"
201
201
cudnn-version :
202
202
type : string
203
- default : " 8.5 .0.96 "
203
+ default : " 8.8 .0.121 "
204
204
trt-version-short :
205
205
type : string
206
206
default : " 8.6.0"
@@ -252,7 +252,7 @@ commands:
252
252
default : " 8.6.0"
253
253
cudnn-version-long :
254
254
type : string
255
- default : " 8.5 .0.96 "
255
+ default : " 8.8 .0.121 "
256
256
steps :
257
257
- run :
258
258
name : Set up python environment
@@ -474,6 +474,7 @@ commands:
474
474
- run : mkdir -p /tmp/artifacts
475
475
- run :
476
476
name : Run core / C++ tests
477
+ no_output_timeout : 15m
477
478
environment :
478
479
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"
479
480
command : |
@@ -1217,7 +1218,7 @@ parameters:
1217
1218
default : " https://download.pytorch.org/whl/cu117"
1218
1219
cudnn-version :
1219
1220
type : string
1220
- default : " 8.5 .0.96 "
1221
+ default : " 8.8 .0.121 "
1221
1222
trt-version-short :
1222
1223
type : string
1223
1224
default : " 8.6.0"
Original file line number Diff line number Diff line change @@ -71,10 +71,10 @@ http_archive(
71
71
http_archive (
72
72
name = "cudnn" ,
73
73
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" ,
76
76
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" ,
78
78
],
79
79
)
80
80
You can’t perform that action at this time.
0 commit comments