[문제해결] 모델 훈련시 불필요한 경고문 해결하기

You are using a CUDA device ('NVIDIA RTX A6000') that has Tensor Cores. To properly utilize them, you should set `torch.set_float32_matmul_precision('medium' | 'high')` which will trade-off precision for performance. For more details, read https://pytorch.org/docs/stable/generated/torch.set_float32_matmul_precision.html#torch.set_float32_matmul_precision
Initializing distributed: GLOBAL_RANK: 0, MEMBER: 1/4

 

 

대충 너 GPU 환경 좋은데 왜 설정 높게 안해놨냐고 따지는 내용이다.

 

 

 

import torch
torch.set_float32_matmul_precision('high')

 

처음에 이렇게 시작하고나면 해당 경고문은 해결된다.

 

 

 

 

 

그외 각종 지저분한 tensorflow 경고문 해결하는 방법은 아래와 같이 숨기기...가 최선인 것으로 보인다.

import os
import tensorflow as tf
os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'  # TensorFlow 경고 숨기기
tf.get_logger().setLevel('ERROR')  # 로그 레벨 조정