728x90 반응형 GPU3 딥러닝 모델 컨테이너 띄우기(wt. GPU) GPU를 활용해 학습을 하는 딥러닝 모델(BERT)을 컨테이너를 띄워 컨테이너 내에서 학습을 할 수 있는지 확인하는 것이 목표이다. 환경) Ubuntu 20.04 nvidia driver: 470 docker: 20.10.14 nvidia-docker: 20.10.14 1. 베이스 이미지 불러오기 아무것도 없는 컨테이너에서 pip, python, anaconda, pytorch ... 등등 모든 것을 새로 빌드하는건 많은 리소스를 요구한다. 도커 허브에서 이미지를 pull 하는 방법으로 과정을 대폭 간소화한다. 1. 별이 가장 많은 miniconda3 이미지를 불러온다. docker pull continuumio/miniconda3 편리하게도 위의 이미지에 컨테이너는 python은 물론 pip, git .. 2022. 3. 31. nvidia-docker 설치 및 컨테이너 띄우기 with GPU GPU를 사용하는 딥러닝 모델이 학습하는 환경을 이미지로 빌드하여 컨테이너 내에서 호스트에서 사용하는 GPU가 작동하는지 확인하는 것이 목표이다. 학습환경 OS: Ubuntu 20.04 Nvidia driver: 470 모델 BERT classfication 우분투, 엔비디아 드라이버가 설치되어 있는 상태로 시작했다. 1. docker 설치 -업데이트 및 패키지 설치 $ sudo apt update $ sudo apt-get install -y ca-certificates \ curl \ software-properties-common \ apt-transport-https \ gnupg \ lsb-release - GPG키와 저장소 추가 $ curl -fsSL https://download.docker.. 2022. 3. 31. GPU 사용 환경 세팅 **운영체제 우분투 20.04** 1. nvidia-smi 설치 2. cuda-toolkit 설치 3. cuDNN 설치 1. nvidia-smi 설치 1.1 현재 사용중인 그래픽카드 확인 그래픽카드 제원을 엿볼 수 있다. lshw -numeric -C display 1.2 사용가능한 nvidia-driver 확인 ubuntu-drivers devices 1.3 권장 드라이버 설치 recommended된 버전 "470" 을 두번째 줄의 명령어로 설치한다. #sudo ubuntu-drivers autoinstall #권장 드라이버로 자동 설치 sudo apt install -y nvidia-driver-470 #450버전의 driver 설치 1.4 재부팅(필수) 1.5 설치 확인 nvidia-smi 2. c.. 2022. 3. 26. 이전 1 다음 728x90 반응형
