From 2df7ccfa1498f5038b15acd50bc9277ad768dcbf Mon Sep 17 00:00:00 2001 From: Someone Serge Date: Tue, 5 Dec 2023 20:03:13 +0000 Subject: [PATCH] python311Packages.torch: enable_language(CUDA) wants to -lcudart_static? --- pkgs/development/python-modules/torch/default.nix | 1 + 1 file changed, 1 insertion(+) diff --git a/pkgs/development/python-modules/torch/default.nix b/pkgs/development/python-modules/torch/default.nix index 5c81450bb112..8ff396cf9cc3 100644 --- a/pkgs/development/python-modules/torch/default.nix +++ b/pkgs/development/python-modules/torch/default.nix @@ -341,6 +341,7 @@ in buildPythonPackage rec { cuda_cccl.dev # cuda_cudart.dev # cuda_runtime.h and libraries cuda_cudart.lib + cuda_cudart.static cuda_cupti.dev # For kineto cuda_cupti.lib # For kineto cuda_nvcc.dev # crt/host_config.h; even though we include this in nativeBuildinputs, it's needed here too