[기타]Docker 이용해서 cuda11.3 설치하기

본 글은 Docker는 설치되어 있다는걸 전제합니다 

 

1. Docker hub로부터 이미지를 가져오자

 

https://hub.docker.com/r/nvidia/cuda/tags

Docker

원하는 걸 하나 선택해봅시다. 저는 A5000에 맞는 cuda 11.3를 설치할 겁니다.

뒤에 이름 보시면 cuda 버젼 뒤에 OS랑 기타 등등이 명시되어있습니다. 

 

 

 저는 요걸루 결정했습니다. 이름을 알았으니 가져와봅시다

 

docker pull nvidia/cuda:11.3.0-devel-ubuntu20.04

docker pull [이미지 이름] 

 

허브로부터 image를 가져오는 명령어입니다. 다른걸 원하시면 docker hub에서 다른 image 이름을넣으시면 됩니다. 

 

잘설치되었으면 container를 만들어봅시다. 

 

 

 

docker run -it --gpus all --name kj -p 3968:22 -p 32609:8888 nvidia/cuda:11.4.0-devel-ubuntu18.04 nvidia-smi

-p 뒤에 붙은건 포트와 관계된건데 서버용이 아니면 생략하셔도 무방합니다 

저는 --name 명령어로 컨테이너의 이름을 cuda로 정해줬어요 

 

$docker start cuda
$docker attach cuda

 

여기까지 실행해주시면 설치한 이미지로 docker container 생성 및 접근이 완료됩니다