AI/AI23 [AI] Jupyter에서 GPU 사용 여부 확인하기 GPU를 사용하면 tensorflow에서 모델을 효과적으로 학습할 수 있습니다. 간단한 수식을 통해 각 텐서와 연산이 어떠한 장치에 할당되었는지를 출력해서 현재 내 컴퓨터에서 GPU를 사용하고 있는지, CPU를 사용하고 있는지를 확인할 수 있습니다. import tensorflow as tf tf.debugging.set_log_device_placement(True) a = tf.constant([ [1,1], [2,2] ]) b = tf.constant([ [5,5], [7,7] ]) c = tf.matmul(a,b) print(c) tf.debugging.set_log_device_placement(False) 이러한 GPU 관련 내용이 나온다면 GPU가 사용되고 있는 것입니다. 다른 방법으로는, .. 2023. 1. 2. [AI] Pytorch 개요 및 GPU 사용 여부 체크하기 PyTorch는 머신러닝 프레임워크 중 하나입니다. Tensorflow도 머신러닝 프레임워크 중 하나죠. PyTorch를 사용하면 GPU 연동을 통해 효율적으로 딥러닝 모델을 개발할 수 있습니다. colab에서도 사용 가능합니다. import torch 다음과 같이 import하여 사용할 수 있습니다. GPU 사용 여부 체크하기 a = [[1,2],[3,4],[5,6]] x = torch.tensor(a) print(x.is_cuda) x = x.cuda() print(x.is_cuda) x = x.cpu() print(x.is_cuda) x.cuda() 라는 명령어로 GPU로 옮기고, cpu()라는 명령으로 CPU로 옮길 수 있습니다. 2022. 12. 22. [AI] Fashion Mnist 데이터셋을 사용한 CNN 저번 글에서는 Fashion Mnist를 사용하여 이미지의 라벨을 추론하는 실습을 했었는데요 2022.12.17 - [AI, 빅데이터/AI] - [AI] Fashion Mnist 데이터셋으로 이미지 분류하기 [AI] Fashion Mnist 데이터셋으로 이미지 분류하기 안녕하세요! 오늘은 교과서처럼 사용되는 데이터셋중 하나인 Fasion MNIST 데이터셋으로 이미지를 분류해볼 예정입니다. 데이터셋에 대한 자세한 내용은 캐글에서 확인하시길 바랍니다. Fashion MNIST foxyprogramming.tistory.com CNN을 하기 전까지의 과정에 대한 내용은 위의 글에서와 마찬가지로 진행합니다. 라이브러리는 추가할 부분이 있습니다. CNN 모델 구성할 때 필요한 라이브러리를 추가합니다. impor.. 2022. 12. 18. [AI] Fashion Mnist 데이터셋으로 이미지 분류하기 안녕하세요! 오늘은 교과서처럼 사용되는 데이터셋중 하나인 Fasion MNIST 데이터셋으로 이미지를 분류해볼 예정입니다. 데이터셋에 대한 자세한 내용은 캐글에서 확인하시길 바랍니다. Fashion MNIST | Kaggle Fashion MNIST An MNIST-like dataset of 70,000 28x28 labeled fashion images www.kaggle.com 바로 시작해볼까요? 먼저 필요한 라이브러리를 불러옵니다. import tensorflow as tf from tensorflow import keras import numpy as np import matplotlib.pyplot as plt 다음으로 Fasion Mnist 데이터를 불러옵니다. fashion_mnist = .. 2022. 12. 17. 이전 1 2 3 4 5 6 다음
