A100_docker

NVIDIA A100

user name : bert
Nvidia NGC docker image

docker pull nvcr.io/nvidia/pytorch:21.12-py3

bert_VE_SR

컨테이너 생성

docker run \
--name bert_VE_SR \
--gpus all \
-it \
-v ${PWD}/VE_SR:/VE_SR \
-v ${PWD}/datasets:/VE_SR/datasets \
nvcr.io/nvidia/pytorch:21.12-py3 /bin/bash

컨테이너 재진입

docker start bert_VE_SR \
&& docker exec -it bert_VE_SR /bin/bash