Skip to content

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

@DarkPocket

Description

@DarkPocket

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

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions