본문 바로가기

Machine Learning

Docker 환경 구성 (huggingface) docker build 명령어

huggingface는 친절하게도 cpu, gpu, tpu 에 따라 docker 파일을 내장되어있다 또한 텐서플로우인지 텐서플로우에 따라나눠서 내장되어있다. 

 

해당폴더에서 docker 파일 실행시킬 때 명령어

 

docker build -t pytorch:latest  ./ 

 

docker build -t 지정이미지이름:태그(버전이라생각하면됌) ./

'Machine Learning' 카테고리의 다른 글