본문 바로가기

카테고리 없음

Pytorch Docker 설치 및 환경설정

1. docker pull pytorch/pytorch:1.5-cuda10.1-cudnn7-runtime

2. docker images --> pytorch/pytorch 확인

3. docker run -it -p 7100-7120:7100-7120 --runtime=nvidia -h shkim -v /data/shkim/:/home/ --name shkim_pytorch pytorch/pytorch:1.5-cuda10.1-cudnn7-runtime /bin/bash