CUDA 8.0 삭제하고 9.0 설치하는법
- TensorFlow 1.7.0 Release
- remove CUDA 8.0
- CUDA Toolkit 9.0 Downloads
- Download cuDNN v7.1.2 (Mar 21, 2018), for CUDA 9.0
- 마무리
TensorFlow 1.7.0 Release
tensorflow 1.7.0 가 최근 릴리즈되서 업데이트를 진행해보려 한다.
기존에 사용하던 버전은 1.4.0 이어서 CUDA 8.0 과 cuDNN 5.0 을 사용하고 있었다.
포스팅 전에 미리 CUDA 설치를 진행하였는데 당연히 최신 버전인 CUDA 9.1 을 설치하고 후회를 하였다.
아직은 CUDA 9.0 만 지원을 하나보다.
install을 진행해야 하는 순서는
- CUDA 8.0 삭제
- CUDA 9.0 설치
- CUDA patch 설치
- cuDNN 7 설치
순으로 진행해야 한다.
remove CUDA 8.0
다음 2줄로 CUDA 를 지울 수 있다.
sudo apt-get --purge remove 'cuda*'
sudo apt-get autoremove --purge 'cuda*'
혹시 cuda 8.0 폴더가 남아있다면, 다음 경로로 들어가 cuda 8.0 폴더를 지워주자.
(cuda 폴더도 같이 지워주자.)
sudo rm -rf /usr/local/cuda-8.0
sudo rm -rf /usr/local/cuda
CUDA Toolkit 9.0 Downloads
위 사이트에 들어가서 cuda deb 파일을 설치하고 다운로드 폴더로 들어가 다음 코드를 실행한다.
Base Installer
cd Download
sudo dpkg -i cuda-repo-ubuntu1604-9-0-local_9.0.176-1_amd64.deb
sudo apt-key add /var/cuda-repo-9-0-local/7fa2af80.pub
sudo apt-get update
sudo apt-get install cuda
cuBLAS 패치파일을 받는다.
패치 2가 있으면 1은 안받아도 되는지 몰라서 필자는 1을 설치하고, 2를 설치하였다.
Patch 1 (Released Jan 25, 2018)
cd Download
sudo dpkg -i cuda-repo-ubuntu1604-9-0-local-cublas-performance-update_1.0-1_amd64.deb
sudo apt-get update
sudo apt-get upgrade cuda
Patch 2 (Released Mar 5, 2018)
cd Download
sudo dpkg -i cuda-repo-ubuntu1604-9-0-local-cublas-performance-update-2_1.0-1_amd64.deb
sudo apt-get update
sudo apt-get upgrade cuda
Download cuDNN v7.1.2 (Mar 21, 2018), for CUDA 9.0
cuDNN 은 버전 상관없이 CUDA 9.0 에서 호환되는 최신 버전을 받으면 된다.
cuDNN v7.1.2 Library for Linux 설치 후.
다운로드 폴더로 들어가 cuDNN 압출파일을 푼다.
이후 안에 있는 폴더를 cuda-9.0 폴더에 이동해주면 되는데,
필자는 터미널 용어가 약하기 때문에 관지자 권한으로 폴더를 실행한 후 복 붙을 진행하였다.
# 관리자 권한 폴더 실행
sudo nautilus
사실 터미널 용어로 진행해도 된다.
sudo mv ~/Download/cuda/* /usr/local/cuda-9.0
마무리
이후 python 에서 import tensorflow as tf 를 실행해보면 문제없이 실행되는 것을 볼 수 있다.