I would have assumed that FORCE_CUDA would attempt to install CUDA in all cases.
I suspect the logic here:
|
BUILD_CUDA = (CUDA_HOME is not None) if torch.cuda.is_available() else FORCE_CUDA |
should be
BUILD_CUDA = FORCE_CUDA or (torch.cuda.is_available() and (CUDA_HOME is not None))
Or perhaps I misunderstood the flag. I couldn't find any documentation for it in the installation guide.
Cheers,
Ash