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

해결책

1. 환경의 CUDA 버전이 11.7 이전인 경우

    conda install -c conda-forge cudatoolkit-dev

2. 환경의 CUDA 버전이 11.8 이후인 경우 (cuda-12.1.1을 알맞게 수정! 여기서 버전 목록 확인 가능)

    conda install -c "nvidia/label/cuda-12.1.1" cuda-nvcc cuda-cudart-dev libcublas-dev libcurand-dev
    conda install -c nvidia cuda-profiler-api

뭔 일임?

딥러닝 코드를 깃헙에서 받아서 돌려보려 할 때, 설치 과정에서

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

같은 에러가 나올 수 있다.
아마도, 내가 겪은 경우에서는, 설치 과정에서 cu 코드를 컴파일할 필요가 있기 때문인데, 컴파일러인 nvcc를 찾을 수 없기 때문이다.
엔비디아 웹사이트에 가서 cuda-toolkit을 설치하면 되지만, conda 가상환경마다 CUDA 버전을 다르게 설정하고 싶기 때문에 글로벌한 무언가의 설치는 안하고 싶다.
CUDA 버전이 11.7 이하인 경우는 cudatoolkit-dev를 다운로드 받으면 되지만, 11.8 이후로는 cudatoolkit-dev가 없어서 필요한 개별 패키지를 모두 다운받아야 한다.

나의 경우는 해결책에 적어놓은 패키지를 설치하니 해결됐는데, 만약에 안된다면 목록에서 필요한 패키지를 더 찾아야한다.