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
728x90
반응형
LIST