tensorflow 를 사용하려면 우선 nvidia 그래픽 드라이버 업데이트, cuda toolkit과 cudnn을 설치를 해야 합니다.
먼저 그래픽 드라이버 버전이 사용하고자 하는 cuda toolkit 버전과 호환되는지 아래 페이지에서 확인해주세요
https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html#major-components
1. cuda toolkit을 설치하기 전 사용하려는 tensorflow 버전의 호환성을 아래 페이지에서 확인합니다.
https://www.tensorflow.org/install/install_sources#tested_source_configurations
2. cuda toolkit download
아래 페이지에서 사용하고자 하는 cuda toolkit을 다운로드 받습니다.
https://developer.nvidia.com/cuda-toolkit-archive
3. cuda toolkit을 다운로드 받은 폴더에서 터미널을 열고
$ sudo sh cuda-8.0.61_375.26_linux.run
입력해서 다운로드 받은 파일을 실행해주세요. 저는 8.0 버전을 다운로드 받았습니다. 버전이 다르다면 다운로드 받은 코드로 수정해서 입력해주세요.
사용약관이 아래처럼 나올텐데, ctrl+c 하여 생략하고
--------------------------
Preface
-------
The following contains specific license terms and conditions
for four separate NVIDIA products. By accepting this
agreement, you agree to comply with all the terms and
conditions applicable to the specific product(s) included
herein.
NVIDIA CUDA Toolkit
그 뒤에 나오는 질문들엔 아래와 같이 입력해주세요
# 저는 이미 nvidia 그래픽 드라이버를 업데이트한 상태이므로 "n"
Install NVIDIA Accelerated Graphics Driver for Linux-x86_64 375.26?
(y)es/(n)o/(q)uit: n
Install the CUDA 8.0 Toolkit?
(y)es/(n)o/(q)uit: y
Enter Toolkit Location
[ default is /usr/local/cuda-8.0 ]:
Do you want to install a symbolic link at /usr/local/cuda?
(y)es/(n)o/(q)uit: y
# sample이 필요하지 않아서 설치하지 않았으므로 "n"
Install the CUDA 8.0 Samples?
(y)es/(n)o/(q)uit: n
설치가 완료되면 아래와 같이 출력됩니다.
===========
= Summary =
===========
Driver: Not Selected
Toolkit: Installed in /usr/local/cuda-8.0
Samples: Not Selected
4. cudnn 설치
https://developer.nvidia.com/rdp/cudnn-archive
아카이브에서 설치한 cuda toolkit과 호환되는 버전을 다운로드 받습니다.
다운로드 받은 폴더에서 터미널을 열고
$ tar xvzf cudnn-8.0-linux-x64-v7.tgz
를 입력하면 압축이 풀리며 폴더가 하나 생성됩니다. cudnn 버전마다 생성되는 폴더 이름은 다릅니다. 저는 cuda 라는 폴더가 생성됬습니다.
생성된 cuda 폴더 안의 include 폴더로 이동하여
$ sudo cp cudnn.h /usr/local/cuda-8.0/include
cuda 폴더 안의 lib64 폴더로 이동하여
$ sudo cp libcudnn* /usr/local/cuda-8.0/lib64
$ sudo chmod a+r /usr/local/cuda-8.0/include/cudnn.h /usr/local/cuda-8.0/lib64/libcudnn*
를 입력합니다.
마지막으로 NVIDIA CUDA Profile Tools Interface 설치를 위해서
$ sudo apt-get install libcupti-dev
를 입력하고 설치해주세요.
5. GPU 활성화
텐서플로우 GPU 버전을 설치했다면 LD_LIBRARY_PATH 와 CUDA_HOME 환경 변수를 지정해야 합니다.
export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/usr/local/cuda-8.0/lib64"
export CUDA_HOME=/usr/local/cuda
↓ 이 글이 도움이 되셨다면 공감을 표시하는 아래 하트를 눌러주세요. 감사합니다.
