nvcc --Version 업데이트 및 torch, pytorch 설치시 nvcc 오류

연구실 GPU 쿠다 마무리 작업 중,

나는 분명 cuda8, cuda9를 설치하였는데, nvcc --version 과 nvcc -V 를 쳤을 때 V7.5로 뜨는 것을 확인했다.

계속 이상하게 생각하던 도중...

torch를 설치 하고 있는데, 다음과 같은 에러가 떴다.

nvcc fatal   : Unsupported gpu architecture 'compute_61'

CMake Error at THC_generated_THCHalf.cu.o.cmake:207 (message):

  Error generating

  /home/lee/torch/extra/cutorch/build/lib/THC/CMakeFiles/THC.dir//./THC_generated_THCHalf.cu.o

lib/THC/CMakeFiles/THC.dir/build.make:560: recipe for target 'lib/THC/CMakeFiles/THC.dir/THC_generated_THCHalf.cu.o' failed

make[2]: *** [lib/THC/CMakeFiles/THC.dir/THC_generated_THCHalf.cu.o] Error 1

make[2]: *** Waiting for unfinished jobs....

nvcc fatal   : Unsupported gpu architecture 'compute_61'

CMake Error at THC_generated_THCTensorSortChar.cu.o.cmake:207 (message):

  Error generating

  /home/lee/torch/extra/cutorch/build/lib/THC/CMakeFiles/THC.dir/generated/./THC_generated_THCTensorSortChar.cu.o 블라블라...

nvcc fatal 문제 이다. nvcc 문제로 보는게 맞겠다.

나는 서버에서 거리가 좀 되고, 모니터는 무한 리부팅 문제를 해결하지 않아서, 원격으로 하기 때문에 display 로 보는 것이 힘들기 때문에 

vi ~/.profile 을 입력

맨 아랫줄에 다음과 같이 경로 설정을 해줬다.

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

닫고 난 뒤, source ~/.profile 을 해준다.

그러면 이제 경로 설정이 되어서 nvcc --version 을 할 경우 cuda 8 버전으로 잘 잡힌다.

연구실 서버 같은 경우 계정이 약 10개 정도 되는데, 일일이 다 해주기는 번거로운 일이다.

그 때, root 계정(superuser)로 접속하여

vi /etc/profile 을 입력

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

그 후 source /etc/profile 해주면 된다.

이렇게 되면 모든 계정에 대해 쿠다를 8로 잡아준다.

cuDNN 같은 경우에는 워낙 자료가 많으므로 생략

bashrc 로 하는 경우도 있는데, 마찬가지로 해주면 된다.

처음에 cuda9를 설치하고, 9 지원 안해주는 프레임워크가 있어서 uninstall 하지 않고 바로 설치했는데, 문제 없이 되는 것 같다.

cuDNN 도 7에서 6으로 바꿔서 설치했다. 마찬가지로 삭제하지 않고, 메뉴얼대로 따라 했을 뿐인데 잘 된다.

결론 : 굳이 삭제할 필요 없음