zdevito / ATen

ATen: A TENsor library for C++11

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

GPU detection fails when CUDA compilation requires CUDA_HOST_COMPILER to be set

skimo-openhub opened this issue · comments

I don't seem to have permission to write to the repo, so I'm attaching the patch here.
Without it, I get "Automatic GPU detection failed. Building for common architectures.".

0001-use-CUDA_HOST_COMPILER-to-detect-installed-GPU.patch.txt

this has been merged in pytorch/aten master and can be closed now :)