huggingface는 친절하게도 cpu, gpu, tpu 에 따라 docker 파일을 내장되어있다 또한 텐서플로우인지 텐서플로우에 따라나눠서 내장되어있다.
해당폴더에서 docker 파일 실행시킬 때 명령어
docker build -t pytorch:latest ./
docker build -t 지정이미지이름:태그(버전이라생각하면됌) ./
'Machine Learning' 카테고리의 다른 글
AI 개발의 스포트라이트 ‘MLOps’ (0) | 2021.04.27 |
---|---|
연관 규칙을 사용한 장바구니 분석 (0) | 2019.09.26 |
군집화 이해 (0) | 2019.09.26 |
Strategy Pattern (0) | 2019.09.26 |