[PyTorch] nvcc가 안될 때 ~/.bashrc 수정해 환경변수 설정하기

하.... 몇시간 째 나를 괴롭히던 nvcc ...................... 
모든 코드를 돌리려고 할 때 무언가가 계속 삐걱거리고 잘 돌아가지 않는다면(특히 setup.py 등을 build 할 때, nvcc~ 어쩌고 ~ 이런 뉘양스를 풍긴다면..)

그건 nvcc 문제일 가능성이 높다. 

특히 비전공자인 필자는 이로 인해 각종 setup.py build 실패와.. apex 설치 실패 등의 난항을 겪었다. (ㅋㅋ)
이 원인은 바로 bashrc에 환경변수를 추가하지 않아서 그렇다...... 따라서 아래 과정대로 꼭 추가해주자. 일단 bashrc를 다음과 같이 열어준다. 
 

$ vi ~/.bashrc

이렇게 열린 .bashrc 맨 끝에 다음 두 줄을 추가해주자. vim으로 수정하면 된다. (참고로 a를 누르면 수정모드로 들어가고, 수정이 끝났으면 esc를 누르고 !wq를 하면 수정 내용이 저장된다) 그리고 추가할 때 자신의 cuda 버전에 맞는 숫자를 기입하면 된다. (아래는 cuda-11.0 기준임) 

$ export PATH="/usr/local/cuda-11.0/bin:$PATH"
$ export LD_LIBRARY_PATH="/usr/local/cuda-11.0/lib64:$LD_LIBRARY_PATH"

마지막으로 위 수정된 내용을 현재 터미널에 다음과 같이 적용한다. 

$ source ~/.bashrc

 
이 곳의 도움을 받았다! 정말 감사합니다 ㅠ
위 과정을 거쳐서 이제 nvcc를 정상적으로 cuda가 물게 된 감격스러운 모습(아래)을 남기고 게시글을 마무리 하겠다.... 

아래 명령어는 nvcc --version 을 입력하면 된다!