Merge pull request #220399 from ConnorBaker/fix/torchvision-set-cuda-compilers
torchvision: no need to specify CUDAHOSTCXX
This commit is contained in:
@@ -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
|
||||
'';
|
||||
|
||||
Reference in New Issue
Block a user