inplace_abn icon indicating copy to clipboard operation
inplace_abn copied to clipboard

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

Open DarkPocket opened this issue 1 year ago • 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

DarkPocket avatar May 14 '23 16:05 DarkPocket