[Tensorflow]우분투에 Tensorflow-gpu 버전 설치하기. 1

이번에 서버에 tensorflow-gpu환경을 구축하면서 배운점을 정리해봄

 

설명은 정말 순정 ubuntu에서 부터 시작

 

먼저 ctrl+alt+t 를 눌러서 터미널 창을 열고 다음과 같은 명령어 실행

 

$ nvidia-smi

명령어를 치면 다음과 같은 화면이 나온다

 

위와 같이 창이 안나오면 NVIDIA 드라이버가 설치가 안되었다는 것!

다음과 같은 링크를 접속해서 드라이버 파일을 다운로드 받는다: 링크[http://www.nvidia.co.kr/Download/index.aspx?lang=kr]

 

 

자신이 장착한 GPU 와 같이 설정하고 검색을 눌러서 다운로드 받는다

[참고로 나는 NVIDIA TITAN xp 버전이였다]

 

그다음 우분투의 blacklist.conf 파일을 편집해야한다

나는 vim 에디터를 즐겨써서 다음과 같은 명령어을 입력했다

 

$ sudo vim /etc/modprobe.d/blacklist.conf

 

 그다음 밑에 다음과 같은 줄을 추가한다

blacklist amd76x_edac #이 줄은 대부분 파일에 이미 적혀져있는 내용
blacklist vga16fb
blacklist nouveau
blacklist rivafb
blacklist nvidiafb
blacklist rivatv

그다음 esc키를 누르고 ":wq"를 하여 저장

 

다음 엔비디아의 모든 패키지 삭제, 충돌될수있는 상황을 없애야함
$ sudo apt-get remove --purge nvidia*
아무래도 디스플레이 드라이버를 설치해야하므로, 디스플레이 매니저를 종료해야 함.우분투에서 GUI와 상관없는 터미널에 접속하기 위해선매사람마다 키가 다를수도있지만 필자는 ctrl+alt+f1을 누르니 접속이 가능해짐[alt+f1~f6 도 된다고 하니 해보셔도 좋을듯]

$ sudo service lightdm stop

이라는 명령어로 디스플레이 매니저 종료

 

그다음 위에서 다운로드 받은 파일을 가져와서 실행 시키자

 

$ chmod +x NVIDIA-Linux-x86-64-361.42.run <- 파일이름은 다를수가 있음

$ sudo ./NVIDIA-Linux-x86-64-361.42.run

 

설치가 끝나면 다시 디스플레이 매니저 실행

 

$ sudo service lightdm start

그럼 자동으로 디스플레이 화면이 돌아온다

[안될시 ctrl + alt + f7 로 변환]

 

설치 완료 검사는 다음과 같은 명령어 실행

 

$ nvidia-smi

$ nvidia-settings

 

정상적으로 실행할시 설치 완료

 

CUDA 그리고 cuDNN은 다음 포스터에서 진행