2018. 8. 27. 19:39 프로그래밍/Machine Learning
tensorflow 설정
근 1년 만에 다시 시작하려니 개발환경 셋팅하는 것도 낯설다.
또 잊기 전에 정리해두자.
현재 개발 환경은
ubuntu 18.04 LTS
nvidia GTX 1080 Ti
1. gpu 드라이버 설정
아래 둘 중에 한가지 방법으로 설정.
1)
$ ubuntu-drivers devices 내 기기 확인
$ sudo apt install nvidia-driver-390 설치
$ nvidia-smi 로 확인
2)
$ add-apt-repository ppa:graphics-drivers/ppa
$ apt update
$ apt install nvidia-390
$ nvidia-smi
2. cuda 9.0
nvidia 사이트에 가서 .deb 파일을 다운받고 설치.
$ dpkg -i cuda-repo-ubuntu1704-9-0-local_9.0.176-1_amd64.deb
$ apt-key add /var/cuda-repo-9-0-local/7fa2af80.pub
$ apt-get install cuda
3. .bashrc 수정
export PATH=$PATH:/usr/local/cuda/bin
export CUDADIR=/usr/local/cuda
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64
4. cudnn 7.0
( https://developer.nvidia.com/rdp/cudnn-archive )
$ tar xvfz cudnn-9.0-linux-x64-v7.tgz
$ cp cuda/include/* /usr/local/cuda/include
$ cp -d cuda/lib/* /usr/local/cuda/lib64
$ chmod a+r /usr/local/cuda/include/cudnn.h
$ chmod a+r /usr/local/cuda/lib64/libcudnn*
patch 파일들을 다 받아서 순서대로
dpkg -i 패키지.deb 로 설치.
5. tesorflow-gpu 1.10.1
텐서플로우 설치.
$ pip3 install tensorflow-gpu
'프로그래밍 > Machine Learning' 카테고리의 다른 글
CNN MNIST (0) | 2017.08.08 |
---|---|
기본 텐서플로우 소프트맥스 MNIST 코드. (0) | 2017.07.24 |