'1080 ti'에 해당되는 글 1건

  1. 2018.08.27 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
Posted by 게 르 니 카
이전버튼 1 이전버튼

블로그 이미지
게 르 니 카

공지사항

Yesterday
Today
Total

달력

 « |  » 2024.5
1 2 3 4
5 6 7 8 9 10 11
12 13 14 15 16 17 18
19 20 21 22 23 24 25
26 27 28 29 30 31

최근에 올라온 글

최근에 달린 댓글

최근에 받은 트랙백

글 보관함