cudaPackages_11_8: init, with support for jetsons#194791
cudaPackages_11_8: init, with support for jetsons#194791SomeoneSerge wants to merge 4 commits intoNixOS:masterfrom
Conversation
| defaultBuild = { "tensorrt" = allBuilds.${computeName tensorRTDefaultVersion}; }; | ||
| in allBuilds // defaultBuild; | ||
| defaultBuildName = computeName tensorRTDefaultVersion; | ||
| defaultBuild = |
There was a problem hiding this comment.
I'm only touching this because of eval errors.
But I'm also really overwhelmed reading this piece
There was a problem hiding this comment.
Fix seems good to me.
We create an attribute tensorr_${version} for every available tensorrt version. Then we add a tensorrt attribute (so without version, however, for cudatoolklit 11.8 there is no compatible tensorrt`, and hence we cannot add a default.
In the future we might need a fix like this for the other extensions as well.
|
also note that there is #194705 open as well |
|
Ouch, missed it, didn't get a notification. I guess I'll wait for the other PR and then poke the jetson support |
| defaultBuildName = computeName tensorRTDefaultVersion; | ||
| defaultBuild = | ||
| if | ||
| builtins.hasAttr defaultBuildName allBuilds then |
There was a problem hiding this comment.
I thought so too, but then I realized I'd have to use it with an interpolation: ? ${defaultBuildName}. Wasn't sure if that's better than hasAttr
Anyway, @ dguibert handles this differently in # 194705 and I'm yet to have looked at it thoroughly
3 participants
Description of changes
CUDA 11.8 has been released, this time including redist packages for jetsons.
This PR adds cuda 11.8 redist packages with "x86_64-linux" and "aarch64-linux" platforms.
This also packages cudnn 8.6.0 (x86_64 only), since 8.4.0 doesn't officially support cuda 11.8.
Jetson support is still very limited in that there's still not redist packages for CUDNN on aarch64-linux (-> we cannot build pytorch).
I haven't meaningfully tested jetson support yet.
Things done
nix build --impure .#cudaPackages_11_8.libcublason platform(s)Notify maintainers
@NixOS/cuda-maintainers