--cuda_ext was requested, but nvcc was not found 에러 해결

apex를 깔다가 raise RuntimeError("--cuda_ext was requested, but nvcc was not found. 이 에러를 만났다.

찾아보니 conda에서 cudatoolkit을 설치하면 subset으로 깔기 때문에 nvcc를 설치하지 않는게 이유였다. 실제로 cuda를 이용해 GPU를 사용하는 코드는 문제없이 돌아가지만 nvcc는 실행이 되질 않았었다.

apt-get으로 설치하면 그래픽 드라이버가 깨져 무한 로그인에 걸릴 수 있으니 조심하자. 그래서 안전하게 nvcc를 추가 설치할 수 있도록 어떤 분이 다음 이슈에 친절하게 답변해주셨다.

Ubuntu Install Error: RuntimeError("--cuda_ext was requested, but nvcc was not found... · Issue #473 · NVIDIA/apex

wget http://developer.download.nvidia.com/compute/cuda/10.1/Prod/local_installers/cuda_10.1.243_418.87.00_linux.run
sudo sh cuda_10.1.243_418.87.00_linux.run --silent --toolkit
echo "export PATH=/usr/local/cuda-10.1/bin:/usr/local/cuda-10.1/NsightCompute-2019.1${PATH:+:${PATH}}" >> .bashrc
echo "export LD_LIBRARY_PATH=/usr/local/cuda-10.1/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}" >> .bashrc
rm cuda_10.1.243_418.87.00_linux.run

나는 다행히 cuda 10.1을 쓰고 있어서 그대로 실행하여 nvcc와 apex를 성공적으로 깔았지만 맞는 cuda 버전을 확인하고 명령어를 수정해서 설치하는 것을 권장한다.