CUDA_VISIBLE_DEVICES
-
Tensorflow 세팅 확인(버전확인, GPU확인, GPU 지정, 지정된 GPU 확인)Data Science/Tensorflow 2022. 10. 24. 21:06
본글은 Tensorflow 설치 및 Cuda 설치를 다루지 않습니다. 두가지 세팅 완료 후 확인 부분을 다루고 있습니다. 1. 라이브러리 Import import tensorflow as tf 2. Tensorflow 버전확인 print(tf.__version__) 3. GPU 상태 확인 "!"를 통해서 터미널에서 사용가능한 명령어를 사용할 수 있다. 터미널(또는 cmd)에서 nvidia-smi 명령어를 통해서 gpu의 상태를 확인할 수 있다. !nvidia-smi 위 그림에서 확인 가능한 GPU정보는 다음과 같다. 현재 1개의 GPU가 0번으로 연결되어 있으며, 해당 GPU는 RTX A5000이다. (만약 여러개의 GPU가 연결되어 있다면 순차적으로 0~n번의 번호를 부여받는다.) 4. 사용할 GPU ..