Tensorflow Docker

홍성우·2020년 4월 21일
0

1. CUDA 이미지 도커

$ docker run --runtime=nvidia -it nvidia/cuda:10.1-cudnn7-devel-ubuntu18.04
$ apt-get update
$ apt-get install python3-pip
$ pip3 install --upgrade pip
$ pip3 install tensorflow-gpu==2.1.0
$ export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64

2. Docker 컨테이너 to 이미지

$ docker ps -a
#이미지로 만들고 싶은 컨테이너명 복사
$ docker commit [컨테이너명]
$ docker images
#방금 만들어진 이미지명 복사
$ docker tag [이미지명] [태그명]

3. Docker 이미지 백업

$ sudo docker save -o [저장할 이름].tar [이미지명]

4. Docker 이미지 복원

$ sudo docker load < [백업한 파일이름].tar

0개의 댓글