1. 그래픽 드라이버를 설치 합니다.
아래는 자동으로 권장 드라이버에 맞는 그래픽 드라이버를 설치 해줍니다.
sudo ubuntu-drivers autoinstall아래는 특정 버전으로 선택하여 설치하는 방법입니다.
sudo apt install nvidia-driver-460아래 명령어를 입력하면 그래픽 카드 및 설치 가능 드라이버를 표시해줍니다.
ubuntu-drivers devices
드라이버가 설치되면 reboot를 진행합니다.
2. Docker를 설치 합니다.
아래의 명령어를 이용해 docker를 설치합니다.
#from https://gist.github.com/nathzi1505/d2aab27ff93a3a9d82dada1336c45041
sudo apt-get update
sudo apt-get remove docker docker-engine docker.io
sudo apt install docker.io
sudo systemctl start docker
sudo systemctl enable docker
docker --version
sudo usermod -a -G docker $USER
newgrp docker
sudo apt install curl
distribution=$(. /etc/os-release;echo $ID$VERSION_ID)
curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add -
curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.list | sudo tee /etc/apt/sources.list.d/nvidia-docker.list
sudo apt-get update && sudo apt-get install -y nvidia-container-toolkit
sudo systemctl restart docker도커가 설치되면 reboot를 진행합니다.
3. Tensorflow docker 를 설치 합니다.
www.tensorflow.org/install/docker?hl=ko
Docker | TensorFlow
Docker는 컨테이너를 사용하여 TensorFlow 설치를 나머지 시스템에서 격리하는 가상 환경을 만듭니다. TensorFlow 프로그램은 호스트 머신과 리소스를 공유(디렉터리 액세스, GPU 사용, 인터넷 연결 등)
www.tensorflow.org
위를 참고해 tensorflow docker를 설치합니다.
저는 tensorflow-gpu 2.3.0를 설치하려 합니다.
sudo docker pull tensorflow/tensorflow:2.3.0-gpu
설치가 완료된 후 아래의 코드를 입력하면
docker images
위에처럼 tensorflow-gpu image가 깔리는 것을 볼 수 있습니다.
아래 명령어로 docker를 실행합니다.
docker run --gpus all -it -v $(pwd):/lushtoner --rm tensorflow/tensorflow:2.3.0-gpu-v 는 마운트로
왼쪽은 docker를 실행하는 현재 리눅스 경로이고
/lushtoner는 도커 환경에서 작업을 수행할 경로로 보면 될듯 합니다.
hello 폴더가 잘 들어가 있는 것을 확인할 수 있습니다.
그럼 위와 같이 Tensorflow가 실행되는 모습을 볼 수 있습니다.
위와 같이 import tensorflow하면 tensorflow도 잘 import되는 것을 알 수 있습니다.
참고
'Language&Framework&Etc > Docker' 카테고리의 다른 글
| Docker 삭제하기 (0) | 2021.12.10 |
|---|---|
| Dockerfile 실행 방법(image 생성) (0) | 2021.03.03 |
| Tensorflow Docker와 Visual Studio Code 연동 (0) | 2021.02.04 |
| Docker에서 Pytorch 설치 (0) | 2021.02.03 |
| Docker 사용법 정리 (0) | 2021.02.02 |
