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
This commit is contained in:
Connor Baker
2023-03-09 18:14:02 -05:00
parent 61fa97e527
commit 0549af80ec
@@ -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
'';