inplace_abn
inplace_abn copied to clipboard
OSError: CUDA_HOME environment variable is not set. Please set it to your CUDA install root
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.
How can I install it?
use git clone https://github.com/mapillary/inplace_abn.git
?