Docker ubuntu18.04 nvidia/cuda, pytorch 등등 세팅하기

1. docker 컨테이너 만들기

// image : https://hub.docker.com/r/nvidia/cuda/ 에서 태그 보고 사용
// -p : 사용할 포트 번호 외부포트(컴퓨터):내부포트(컨테이너), -를 사용하면 범위 지정 가능

docker start -itd -p 5554:22 -p 5555-5556:5555-5556 --name 컨테이너이름 nvidia:10.1-cudnn7-runtime-ubuntu18.04 bash
docker exec -it 컨테이너이름 or 컨테이너id bash //컨테이너 bash 실행 연결

2. 컨테이너 우분투 세팅

passwd root    //root 패스워드  설정
apt update && apt upgrade
apt install vim
vim /etc/apt/sources.list
:%s/\/\/.*ubuntu.com/\/\/mirror.kakao.com/g    //서버 주소 카카오로 변경하면 다운로드 속도 향상
:wq
apt install openssh-server
vim /etc/ssh/sshd_config
port 22
ListenAddress 0.0.0.0
PermitRootLogin yes
:wq
service ssh start
// pytorch 설치, https://pytorch.org/
apt install python3-pip
pip3 install torch torchvision