Skip to content

Commit e8a6ee8

Browse files
Fix cudastack for JetPack 5. (#1549)
My JetPack 5 has CUDNN 8.6.0 installed. ``` dpkg -l | grep cudnn ii libcudnn8 8.6.0.166-1+cuda11.4 arm64 cuDNN runtime libraries ii libcudnn8-dev 8.6.0.166-1+cuda11.4 arm64 cuDNN development libraries and headers ii libcudnn8-samples 8.6.0.166-1+cuda11.4 arm64 cuDNN samples ```
1 parent 6d45377 commit e8a6ee8

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

packages/cuda/cudastack/config.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -54,8 +54,8 @@ def cuda_stack_args():
5454
tensorrt_url = f"{tensorrt_base_url}/10.0.1/tars/TensorRT-10.0.1.6.l4t.aarch64-gnu.cuda-12.4.tar.gz"
5555
nccl_ver = '2.27.7'
5656
else: # JetPack 5
57-
cudnn_ver = '8.9.4'
58-
cudnn_url = "https://nvidia.box.com/shared/static/ht4li6b0j365ta7b76a6gw29rk5xh8cy.deb"
57+
cudnn_ver = '8.6.0'
58+
cudnn_url = "https://repo.download.nvidia.com/jetson/common/pool/main/c/cudnn/libcudnn8_8.6.0.166-1+cuda11.4_arm64.deb"
5959
cudnn_packages = "libcudnn8 libcudnn8-dev"
6060
tensorrt_ver = '8.6.0'
6161
tensorrt_url = "https://nvidia.box.com/shared/static/hmwr57hm88bxqrycvlyma34c3k4c53t9.deb"
@@ -170,13 +170,13 @@ def cuda_stack(name, with_tensorrt=False, minimal=False, requires=None):
170170
cuda_stack('cudastack:minimal',
171171
with_tensorrt=False,
172172
minimal=True,
173-
requires='>=36'),
173+
requires='>=35'),
174174

175175
# Standard: + TensorRT and all CUDA libraries
176176
cuda_stack('cudastack:standard',
177177
with_tensorrt=True,
178178
minimal=False,
179-
requires='>=36'),
179+
requires='>=35'),
180180
]
181181

182182
elif IS_SBSA and IS_CONFIG:

0 commit comments

Comments
 (0)