mapillary/inplace_abn

OSError: CUDA_HOME environment variable is not set. Please set it to your CUDA install root

Opened this issue · 1 comments

Ubuntu 22.04.2 LTS (GNU/Linux 5.15.0-56-generic x86_64), no GPU.
requirements.txt
torch==1.13.1
torchvision==0.14.1
tqdm==4.65.0
typing_extensions==4.5.0
uc-micro-py==1.0.2
urllib3==2.0.2
uvicorn==0.22.0
websockets==11.0.3
win32-setctime==1.1.0
yarl==1.9.2
zipp==3.15.0

run python3 -m pip install inplace-abn==1.1.0 have error.

  File "/home/ubuntu/.local/share/virtualenvs/model_ML-Danbooru-ekWEn069/lib/python3.10/site-packages/torch/utils/cpp_extension.py", line 2230, in _join_cuda_home
          raise EnvironmentError('CUDA_HOME environment variable is not set. '
      OSError: CUDA_HOME environment variable is not set. Please set it to your CUDA install root.

image

How can I install it?
use git clone https://github.com/mapillary/inplace_abn.git

Same error over here using PyTorch 2.1.0 and TorchVision 0.16.0 with GPU.