DeepFlowest

[Docker] Ubuntu18.04에서 CUDA 10.2, cudnn 7.6.5, nvidia driver 설치 본문

Docker

[Docker] Ubuntu18.04에서 CUDA 10.2, cudnn 7.6.5, nvidia driver 설치

Orange57 2020. 12. 8. 10:32
728x90
반응형
SMALL
  • Ubuntu 18.04
  • CUDA 10.2
  • cudnn 7.6.5
  • OpenCV 3.4.2

1. 원하는 OS 환경 이미지 pull

$ docker pull nvidia/cuda:10.2-cudnn7-devel-ubuntu18.04

2. 이미지가 잘 설치되었나 확인

$ docker images

3. 실행

$ docker run -it --name cuda_10 nvidia/cuda:10.2-cudnn7-devel-ubuntu18.04

4. CUDA, cudnn 버전 확인

# CUDA 버전 확인
$ nvcc --version

# cudnn 버전 확인
$ cat /usr/include/x86_64-linux-gnu/cudnn_v*.h | grep CUDNN_MAJOR -A 2

 

5. Nvidia Driver 설치

[사전 설치]

1) lshw 설치
$ apt-get install lshw

2) ubuntu-drivers 설치
$ sudo apt-get install ubuntu-drivers-common

3) add-apt-repository 설치
$ sudo apt-get install software-properties-common

 

nvidia driver 설치 참고 사이트

728x90
반응형
LIST
Comments