Unverified Commit dc30654a authored by Samuel Ainsworth's avatar Samuel Ainsworth Committed by GitHub
Browse files

Merge pull request #220399 from ConnorBaker/fix/torchvision-set-cuda-compilers 

torchvision: no need to specify CUDAHOSTCXX
parents 5e41f124 0549af80

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

  '';