Open Saafke opened 2 years ago
What is the version of cuda under CUDA_HOME
?
Hmm, if I do echo $CUDA_HOME, it prints an empty line
[BTW - I can run detectron2 succesfully in this environment]
It seems that pytorch will find the potential path of CUDA_HOME itself. And if it is None, the compling will raise error. So I think it is not the key point of this error.
if CUDA_HOME is None:
raise NotImplementedError('CUDA_HOME is None. Please set environment variable CUDA_HOME.')
Hi,
I successfully followed the installation instructions in INSTALL.md, namely:
However, when running the demo I get the following:
From a web search, it seems that this error occurs when the wrong CUDA is installed. However, I correctly installed cudatoolkit 11.1 from the installation procedure above. What else could be the issue?
FYI - the demo runs fine if I run it on my CPU (using the MODEL.DEVICE cpu)