GPU

python 메모

[GPU] pynvml 모듈로 gpu 사용량 체크하기

GPU 모니터링 툴이 별도로 없을 때 pynvml을 사용하여 GPU 사용량 정보 등을 로그로 남길 수 있는 방법을 정리하고자 한다. 1. GPU 정보 확인하기 import pynvml print("Driver Version:", pynvml.nvmlSystemGetDriverVersion()) >>>> Driver Version: 4xx.1xx.xx 2. GPU 사용량 체크하기 GPU가 여러 장 있을 때 GPU별 utilization, memory 사용량 평균을 다음과 같이 기록해둘 수 있다. from threading import Thread import numpy as np import torch .... if __name__ == "__main__": def schedule_gpu_memory_lo..

개발

[ubuntu] GPU 사용 모니터링

실험 돌릴 때 GPU 사용량을 보기 위해 nividia-smi + enter를 무한히 반복했었다.. 다른 분이 알려준 방법을 적어두려고 한다 Watch를 사용한 GPU 사용 모니터링 watch -n 1 nividia-smi를 사용하면 GPU 사용량을 화면에 1초마다 찍어주며, 아주 편리하다. 5초마다 사용량을 찍고 싶으면 -n 5를 해주면 된다. watch -n 1 nvidia-smi

개발

tensorflow GPU 확인 방법

tensorflow-gpu 버전을 설치한 후 실행환경에서 GPU를 잡고 있는지 확인할 수 있는 방법을 정리해보겠습니다. 1) tensorflow 내에서 확인 ># python3 >>> from tensorflow.python.client import device_lib >>> device_lib.list_local_devices() ... 아래와 같이 GPU가 뜨는 것을 확인할 수 있습니다. 2) nvidia-smi로 확인 terminal 창에서 nvidia-smi를 입력하면 python3 process가 GPU를 잡고 있는 것을 확인할 수 있습니다.

개발

Ubuntu 18.04 GPU 개발환경 세팅 (3) - CUDA 삭제 및 재설치

다 설치한 후 python3에서 아래 명령어로 GPU device가 tensorflow에서 잘 불러와지는지 확인해봤습니다. [tensorflow에서 GPU 확인하는 간단한 방법] sess = tf.Session(config=tf.ConfigProto(log_device_placement=True)) # python3 Python 3.7.5 (default, Apr 5 2020, 15:53:27) [GCC 7.5.0] on linux Type "help", "copyright", "credits" or "license" for more information. >>> import tensorflow as tf >>> tf.__version__ '1.14.0' >>> sess = tf.Session(confi..

개발

Ubuntu 18.04 GPU 개발환경 세팅 (2)

GPU 드라이버 설치한 후 tensorflow 개발 환경을 구축해보려고 합니다. 설치환경은 다음과 같아요 운영 체제(OS): Ubuntu 18.04 그래픽 카드(GPU): GeForce RTX 2060 super CUDA 10.0 cuDNN 7.4 Python 3.7 tensorflow-gpu-1.14 1. NVIDIA 그래픽 드라이버 확인 nvidia-smi -q | grep Product 명령어로 GPU 드라이버 설치된 것을 확인해볼 수 있습니다. $ nvidia-smi -q | grep Product Product Name : GeForce RTX 2060 SUPER Product Brand : GeForce 2. CUDA 설치 CUDA, cuDNN을 설치해줘야 합니다. tensorflow, ten..

개발

Ubuntu 18.04 GPU 개발환경 세팅 (1)

GPU를 받았습니다. 컴퓨터 선이 케이블로 정리가 빡세게 되어있어서 선 다 끊고 메인보드에 연결하고 전원선 연결한 후 드라이버를 설치했습니다. Ubuntu: LTS 18.04 GPU: GEFORCE RTX 2060 super 설치하는 방법에는 크게 두 가지가 있습니다. (1) 드라이버를 직접 찾고 원하는 버전을 설치하는 방법과 (2) 자동으로 설치해주는 방법이 있는데 둘 다 소개하고자 합니다. 1. 드라이버 검색 및 지정하여 설치 그래픽카드 정보, 드라이버 정보 확인 아래 명령어로 그래픽 카드 정보를 확인해보겠습니다. $ lshw -C display WARNING: you should run this program as super-user. *-display description: VGA compatib..

Fine애플
'GPU' 태그의 글 목록