From 0549af80ecf15c2d75e4163307c9a2de92c4bc83 Mon Sep 17 00:00:00 2001 From: Connor Baker Date: Thu, 9 Mar 2023 18:14:02 -0500 Subject: [PATCH] torchvision: no need to specify CUDAHOSTCXX CUDAHOSTCXX is defined for us in https://github.com/NixOS/nixpkgs/blob/a59944f1c11a5e1a6bd33cad379dae257d4bf40b/pkgs/development/compilers/cudatoolkit/redist/overrides.nix#L50 --- pkgs/development/python-modules/torchvision/default.nix | 1 - 1 file changed, 1 deletion(-) diff --git a/pkgs/development/python-modules/torchvision/default.nix b/pkgs/development/python-modules/torchvision/default.nix index a9c5defbe38e..8201d9c72b73 100644 --- a/pkgs/development/python-modules/torchvision/default.nix +++ b/pkgs/development/python-modules/torchvision/default.nix @@ -68,7 +68,6 @@ buildPythonPackage { + lib.optionalString cudaSupport '' export CC=${cudatoolkit.cc}/bin/cc export CXX=${cudatoolkit.cc}/bin/c++ - export CUDAHOSTCXX=${cudatoolkit.cc}/bin/c++ export TORCH_CUDA_ARCH_LIST="${gpuTargetString}" export FORCE_CUDA=1 '';