Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

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

Open
DarkPocket opened this issue May 14, 2023 · 1 comment

Comments

@DarkPocket
Copy link

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

@santiagorg2401
Copy link

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

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants