반응형 컨테이너 생성1 [Docker] - windows환경에서 pytorch 컨테이너 생성, 재시작 및 폴더 마운트 # 파이토치 이미지 선택 도커허브에서 pytorch 검색 후 여러 이미지 중 'pytorch/pytorch'를 들어가 Tags를 누르면 그 안에서 여러 이미지를 또 검색할 수 있다. 기존에 개발하던 환경과 동일한 cuda 10.2 버전을 선택해서 pull 하였다. # 컨테이너 생성 - gpu버전으로 생성 1. gpu 버전 생성 run은 새로운 컨테이너를 생성하고 실행하는 명령어, run으로 이미 생성된 컨테이너에 연결할 수 없다 다음과 같이 '--gpus all'을 붙여야 gpu를 사용할 수 있다. * nvidia-smi -> 사용하고 있는 2개의 gpu가 인식된다. * torch.cuda.is_available() -> True 로 나온다. 2. non-gpu 버전 생성 docker run -it --.. 2021. 12. 21. 이전 1 다음 반응형
