카테고리 없음
Pytorch Docker 설치 및 환경설정
바나나인간
2020. 6. 15. 17:15
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