우분투에서 패키지 설치 전에는 항상 패키지를 최신으로 만들어주셔야합니다.
해당 명령어를 통해 항상 최신으로 업데이트 해주세요.
1
2
3 |
sudo apt-get install software-properties-common
sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt-get update |
완료되면 왼쪽 탭 메뉴에 톱니바퀴 모양의 아이콘으로 되어있는 System Settings(시스템 설정)에 들어가서 Software & Updates(소프트웨어 & 업데이트) -> Additional Drivers(추가 드라이버) 탭으로 이동해주세요.
CUDA 8.0은 최소 367.4x 이상의 드라이버가 필요하다고 합니다. 저는 384로 설치했습니다.
같이 선택한 후 Apply Changes(바뀐 내용 적용) 버튼을 클릭해 설치해주세요. 설치가 끝나면 reboot(재시작)을 해주세요.
재부팅이 완료되고 터미널에서
이제 CUDA ToolKit을 설치하시면 됩니다. 제가 설치한 버전은 8.0입니다. (저도 블로그 참고 하였습니다.)
CUDA Download에서 회원가입을 하고 로그인을 한 뒤
Select Target Platform 메뉴에서 Linux -> x86_64 -> Ubuntu -> 16.04 -> runfile(local)을 다운받으세요.
다운이 완료되고 경로로가서 runfile을 실행합니다.
1
2
3
4
5
6
7
8
9
10
11
12
13
14 |
accept/decline/quit: accept
이후엔 옵션을 설정하는 부분으로
Install NVIDIA Accelerated Graphics Driver for Linu….?이 질문은 이미 위에서 설치했기 때문에
(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 ]:
엔터를 쳐서 default를 사용한다.
Do you want to install a symbolic link at /usr/local/cuda?
(y)es/(n)o/(q)uit: y
Install the CUDA 8.0 Samples?
(y)es/(n)o/(q)uit: y
Enter CUDA Samples Location [ default is /home/사용자 이름 ]:
엔터를 쳐서 default를 사용한다. |
설치가 진행되고 끝나면 Summary를 통해 결과를 확인할 수 있습니다. 우리가 확인해야할 Summary는
1
2
3
4 |
export PATH=/usr/local/cuda-8.0/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64\${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
source ~/.bashrc
nvcc --version |
이렇게 출력되면 됩니다.
1
2
3
4 |
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2016 NVIDIA Corporation
Built on Sun_Sep__4_22:14:01_CDT_2016
Cuda compilation tools, release 8.0, V8.0.44 |
마지막으로 cuDNN 5.1을 설치해주세요.
cuDNN Download 에서 cuDNN v5.1과 CUDA 8.0에 해당하는 버튼을 누르고
cuDNN v5.1 Library for Linux를 클릭해 다운받아주세요.
1
2
3
4
5
6
7 |
cd <다운로드 경로>
sudo tar -xzvf cudnn-8.0-linux-x64-v5.1.tgz
cd cuda
sudo cp include/cudnn.h /usr/local/cuda/include
sudo cp lib64/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2 |
이때 출력 화면에 CUDNN_MAJOR가 포함되어 있으면 제대로 설치된 것입니다.
'IT > YOLO' 카테고리의 다른 글
[6] Ubuntu 16.04 APM(Apache2, PHP, Mysql) 설치 (0) | 2018.06.25 |
---|---|
[5] YOLO 데이터 학습 (51) | 2018.06.16 |
[4] Ubuntu 16.04 YOLO 설치 (13) | 2018.06.16 |
[3] Ubuntu 16.04 Opencv3.2 설치 (2) | 2018.06.16 |
[1] 프로그램 개발 배경 (0) | 2018.06.15 |
포스팅이 좋았다면 "좋아요❤️" 또는 "구독👍🏻" 해주세요!