diff options
author | Connor Baker <connor.baker@tweag.io> | 2023-03-18 16:43:42 -0400 |
---|---|---|
committer | Connor Baker <connor.baker@tweag.io> | 2023-03-18 16:43:42 -0400 |
commit | aacbb6d021d617881630c27088f311a5bc0e2145 (patch) | |
tree | c16e8719c84b6bc4c3acf79331a2fc5ecd4101eb /pkgs/top-level/python-packages.nix | |
parent | 7913ebc8c3f1051732d21309e9534d89d16c02db (diff) |
tiny-cuda-nn: init 1.6
Diffstat (limited to 'pkgs/top-level/python-packages.nix')
-rw-r--r-- | pkgs/top-level/python-packages.nix | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/pkgs/top-level/python-packages.nix b/pkgs/top-level/python-packages.nix index de3f2ba61f0fe..def6dd5472404 100644 --- a/pkgs/top-level/python-packages.nix +++ b/pkgs/top-level/python-packages.nix @@ -11694,6 +11694,12 @@ self: super: with self; { timm = callPackage ../development/python-modules/timm { }; + tiny-cuda-nn = toPythonModule (pkgs.tiny-cuda-nn.override { + cudaPackages = self.torch.cudaPackages; + python3Packages = self; + pythonSupport = true; + }); + tinycss2 = callPackage ../development/python-modules/tinycss2 { }; tinycss = callPackage ../development/python-modules/tinycss { }; |