pyrox.dev
lol
python3Packages.torchaudio: fix the cuda build
+16
-3
pkgs/development/python-modules/torchaudio/default.nix
+16
-3
pkgs/development/python-modules/torchaudio/default.nix
···
6
, ninja
7
, pybind11
8
, torch
9
-
, cudaSupport ? false
10
, cudaPackages
11
}:
12
···
27
--replace "_fetch_archives(_parse_sources())" "pass"
28
'';
29
30
nativeBuildInputs = [
31
cmake
32
pkg-config
33
ninja
34
] ++ lib.optionals cudaSupport [
35
-
cudaPackages.cudatoolkit
36
];
37
buildInputs = [
38
pybind11
39
] ++ lib.optionals cudaSupport [
40
-
cudaPackages.cudnn
41
];
42
propagatedBuildInputs = [
43
torch
···
6
, ninja
7
, pybind11
8
, torch
9
+
, cudaSupport ? torch.cudaSupport
10
, cudaPackages
11
}:
12
···
27
--replace "_fetch_archives(_parse_sources())" "pass"
28
'';
29
30
+
env = {
31
+
TORCH_CUDA_ARCH_LIST = "${lib.concatStringsSep ";" torch.cudaCapabilities}";
32
+
};
33
+
34
nativeBuildInputs = [
35
cmake
36
pkg-config
37
ninja
38
] ++ lib.optionals cudaSupport [
39
+
cudaPackages.cuda_nvcc
40
];
41
buildInputs = [
42
pybind11
43
] ++ lib.optionals cudaSupport [
44
+
cudaPackages.libcurand.dev
45
+
cudaPackages.libcurand.lib
46
+
cudaPackages.cuda_cudart # cuda_runtime.h and libraries
47
+
cudaPackages.cuda_cccl.dev # <thrust/*>
48
+
cudaPackages.cuda_nvtx.dev
49
+
cudaPackages.cuda_nvtx.lib # -llibNVToolsExt
50
+
cudaPackages.libcublas.dev
51
+
cudaPackages.libcublas.lib
52
+
cudaPackages.libcufft.dev
53
+
cudaPackages.libcufft.lib
54
];
55
propagatedBuildInputs = [
56
torch