728x90 반응형 nVidia4 728x90 반응형 도커에서 텐서플로우 GPU를 사용하기 위한 환경설정 텐서플로우를 사용하기 위한 환경 설정은 여러가지 방법이 있지만 개발된 딥러닝 알고리즘에 따라 사용되는 버전과 종속성이 다양하기 때문에 도커 환경에서 설치하면 쉽게 버전별 관리가 가능합니다. 오늘은 도커 환경에서 텐서플로우를 사용하기 위한 환경설정방법을 알아보도록 하겠습니다. 도커를 이용해서 텐서플로우를 설치할 때 특징은 아래와 같습니다. CUDA 툴킷을 설치할 필요없이 호스트에 Nvidia GPU 드라이버만 있도 사용이 가능합니다. 즉, 도커에는 Tensorflow와 그 버전에 맞는 CUDA 환경이 이미 갖추어져 있습니다. 텐서플로우 버전에 맞는 이미지를 다운받아서 사용하므로 버전종속성 문제를 해결할 수 있습니다. 다만, 호스트에 Docker 가 설치되어 있어야 합니다.(Docker를 설치) * 도커 사.. 2022. 1. 29. [ YOLO ] Windows 버전 설치 환경 구성하기 YOLO를 윈도우즈 환경에서 구동하기 위한 개발환경을 구성해본다. 먼저 필자의 하드웨어 구성은 아래와 같다. CPU: i7-8700 3.2GHz RAM: 16GB OS: Windows 10 64비트 운영체제 Graphics: Geforce GTX 1070 8GB 개발환경을 구축하기 위한 SW 버전들은 서로 호환성이 있는 버전들도 구성해야 한다. 필자가 구성한 SW의 개발 버전은 아래와 같다. Vusial Studio 2017 CUDA 11.3 cuDNN 8.3 OpenCV 3.14 1. Visual Studio 설치 다운로드는 아래 경로에서 할 수 있다. https://my.visualstudio.com/Downloads?q=visual%20studio%202017&wt.mc_id=o~msft~vscom.. 2021. 6. 17. [ OpenCV ] Ubuntu 18.04 에 CUDA11.1 + CUDNN 8.2.1 설치하기 CUDA 설치 먼저 NVIDIA 그래픽 보드가 있어야 하며, 드라이버가 설치되어 있어야 한다. 드라이버를 설치하는 방법은 아래 참고 2020.11.28 - [관심분야/임베디드] - [리눅스] Ubuntu에서 nVidia 드라이버 설치 [리눅스] Ubuntu에서 nVidia 드라이버 설치 내 하드웨어에 설치된 드라이버를 찾습니다. $ sudo lshw -C display description: VGA compatible controller product: GK107M [GeForce GT 650M] vendor: NVIDIA Corpora.. swiftcam.tistory.com 아래 사이트에서 최신 CUDA 11.1 이상의 최신 버전을 다운 받는다. 자신의 환경과 맞는 파일을 선택하여 다운로드한다. ht.. 2021. 6. 13. Ubuntu 22.04에 NVIDIA 그래픽 드라이버 설치 터미널에서 내 하드웨어에 설치된 드라이버를 찾습니다. $ sudo lshw -C display oem@swift:~/workspace/code/vscode$ sudo lshw -c display [sudo] password for oem: *-display description: VGA compatible controller product: GP104 [GeForce GTX 1070] vendor: NVIDIA Corporation physical id: 0 bus info: pci@0000:01:00.0 logical name: /dev/fb0 version: a1 width: 64 bits clock: 33MHz capabilities: pm msi pciexpress vga_controller bu.. 2020. 11. 28. 이전 1 다음
