-
Notifications
You must be signed in to change notification settings - Fork 560
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Added TensorRT v8.0.1 #4347
Added TensorRT v8.0.1 #4347
Conversation
@maleadt Will a cuda label on this PR allow the download from Nvidia's secured servers? |
This needs separate entries for CUDA 11.0+: https://developer.nvidia.com/compute/machine-learning/tensorrt/secure/8.2.3.0/tars/tensorrt-8.2.3.0.linux.x86_64-gnu.cuda-11.4.cudnn8.2.tar.gz Also, why are you using TensorRT 8.0 while there's 8.2 Update 2 already? I can put the TensorRT tarballs on the Yggdrasil server, if you want. |
What do you mean here? To do multiple builds for each cuda version like in the CUDNN recipe?
My intent is to add recipies for all GA-versions of TensorRT from 8.0 to 8.2 Update 2 due to:
That would be great - feel free to add all GA-versions if you think the aim above makes sense :-) |
Yes, if there's separate builds for different CUDA versions we need to take those into account. |
699f3f9
to
f293478
Compare
I've uploaded the following:
The job currently fails because of incorrect capitalization of these filenames in the recipe. |
f293478
to
fb82e67
Compare
6856157
to
06f6b9d
Compare
@maleadt Is something missing in CUDNN and CUDA_loader ? |
Ah - I probably just need to restrict compat for the CUDNN dependency based on platform... |
From the NVIDIA Jetson APT repo.
b3497c3
to
5cafb63
Compare
5cafb63
to
90ee1c8
Compare
|
Looks ready :-) |
@maleadt good to go? 🙂 |
From my side: Yes - if it's ok with the |
@maleadt Hmm... the CI (on Julia 1: 1.7.2 on x86_64-linux) for the General registry failed: JuliaRegistries/General#56696 |
@giordano @maleadt : Any ideas? Importing the TensorRT JLL fails on first
Importing/loading CUDNN_jll prior to TensorRT_jll does not seem to have an effect. |
The library product should probably be not dlopened automatically? https://docs.binarybuilder.org/stable/reference/#BinaryBuilderBase.LibraryProduct |
Added NVIDIA TensorRT - adapted from CUDNN build script.