Boostcamp AI tech 3기

kaggle 등에서 gpu 확실하게 사용하기 코드 gpu options

mcdn 2022. 11. 19. 15:41
반응형
import numpy as np 
import pandas as pd
import os
from sklearn.model_selection import train_test_split
from sklearn.metrics import confusion_matrix, accuracy_score
import tensorflow as tf
from tensorflow.compat.v1 import ConfigProto
from tensorflow.compat.v1 import InteractiveSession


config = ConfigProto()
config.gpu_options.allow_growth = True
session = InteractiveSession(config=config)

 

 

gpu 할당하고 

이거까지 해야 백프로로 사용한다 (경험상)

 

 

반응형