Commit 0549af80 authored by Connor Baker's avatar Connor Baker
Browse files

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
parent 61fa97e5

pkgs/development/python-modules/torchvision/default.nix

+0 −1
Original line number Diff line number Diff line
@@ -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

  '';