CUDA_HOME environment variable is not set. Please set it to your CUDA install root for pytorch cpp extensions

1. Probelm Description

- Mostly, this error comes from not setting path for nvcc  
- so first check the nvcc using `nvcc --version` 
- nvidia-smi is not related 

# nvidia-smi command works fine
$ nvidia-smi

# nvcc command not found 
$ nvcc --version 

2. Solution

- setting correct path in ~/.bashrc or ~/.zshrc depending on shell you are using 

$ vi ~/.zshrc

# adding below at the end of script 
# export PATH="/usr/local/{installed cuda-version}/bin:$PATH"
# export LD_LIBRARY_PATH="/usr/local/{installed cuda-version}/lib64:$LD_LIBRARY_PATH"
export PATH="/usr/local/cuda-11.0/bin:$PATH"
export LD_LIBRARY_PATH="/usr/local/cuda-11.0/lib64:$LD_LIBRARY_PATH"

$ source ~/.zshrc

1. 문제 상황

- nvcc 경로를 설정하지 않으면 발생 
- `nvcc --version` 를 사용해서 설정 check (설정이 안되어 있으면 command를 못찾음) 
- nvidia-smi 같은 경우 작동함 

# nvidia-smi command works fine
$ nvidia-smi

# nvcc command not found 
$ nvcc --version 

2. Solution

- 사용하는 shell 에 따라 ~./bashrc 또는 ~./zshrc 에서 경로 지정 

$ vi ~/.zshrc

# adding below at the end of script 
# export PATH="/usr/local/{installed cuda-version}/bin:$PATH"
# export LD_LIBRARY_PATH="/usr/local/{installed cuda-version}/lib64:$LD_LIBRARY_PATH"
export PATH="/usr/local/cuda-11.0/bin:$PATH"
export LD_LIBRARY_PATH="/usr/local/cuda-11.0/lib64:$LD_LIBRARY_PATH"

$ source ~/.zshrc

ref

• [https://bo-10000.tistory.com/73]
• [https://github.com/conda/conda/issues/7757]
• [[https://gist.github.com/Brainiarc7/470a57e5c9fc9ab9f9c4e042d5941a40
• (https://gist.github.com/Brainiarc7/470a57e5c9fc9ab9f9c4e042d5941a40%5D)