Tensorflow Docker와 Visual Studio Code 연동

도커 설치 및 Tensorflow gpu 세팅

1. 그래픽 드라이버를 설치 합니다. 아래는 자동으로 권장 드라이버에 맞는 그래픽 드라이버를 설치 해줍니다. sudo ubuntu-drivers autoinstall 아래는 특정 버전으로 선택하여 설치하는 방법입니다. sudo

better-tomorrow.tistory.com

우분투에서 위 글에서 설치한 Tensorflow-gpu와 visual studio code를 연결하는 방법을 알아보겠습니다.

Visual Studio Code 설치

ubuntu software 프로그램을 엽니다.

visual studio code를 검색하여 설치합니다.

저는 Remote-Containers & Python 추가로 및 Python for VSCode, Python Extension Pack 설치해줬습니다.

이제 터미널을 열어 docker를 실행해줍니다.

docker run --gpus all -v /home:/username -it --rm tensorflow/tensorflow:2.3.0-gpu

docker ps로 현재 실행 중인 도커를 아래와 같이 확인할 수 있습니다.

VS code를 실행합니다.

Ctrl + Shift + P 를 눌러 command를 실행합니다

여기서 Remote-containers : Attach to running container... 를 클릭합니다.

그러면 현재 실행 중인 docker container가 나오며 이 중 원하는 container를 실행하면 됩니다.

아래 초록색 영역을 보면 tensorflow/tensorflow-gpu:2.3.0이 잘 연결된 것을 볼 수 있습니다.

Desktop에 test_docker 폴더를 생성해주고 vs code로 폴더를 열어주겠습니다.

test_docker 폴더에 test.py 파일을 생성하고

파이썬을 테스트 해보겠습니다.

Run에 Run without debugging(ctrl + F5)을 클릭하면 아래와 같이 정상적으로 동작하는 것을 확인할 수 있습니다.