about summary refs log tree commit diff
path: root/pkgs/development/python-modules/torch
diff options
context:
space:
mode:
Diffstat (limited to 'pkgs/development/python-modules/torch')
-rw-r--r--pkgs/development/python-modules/torch/default.nix2
1 files changed, 1 insertions, 1 deletions
diff --git a/pkgs/development/python-modules/torch/default.nix b/pkgs/development/python-modules/torch/default.nix
index 5e0c56800073d..98116052649ad 100644
--- a/pkgs/development/python-modules/torch/default.nix
+++ b/pkgs/development/python-modules/torch/default.nix
@@ -1,5 +1,5 @@
 { stdenv, lib, fetchFromGitHub, buildPythonPackage, python,
-  cudaSupport ? false, cudaPackages, magma,
+  config, cudaSupport ? config.cudaSupport or false, cudaPackages, magma,
   useSystemNccl ? true,
   MPISupport ? false, mpi,
   buildDocs ? false,