Machine Learning

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

조잘거리지만 차분한 데이터쟁이 2020. 8. 26. 00:23

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

 

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

 

docker build -t pytorch:latest  ./ 

 

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