728x90 반응형 deeplearninig1 딥러닝 모델 컨테이너 띄우기(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. 이전 1 다음 728x90 반응형