google colab

GPU 변경 → T4 GPU

import torch

# torch 버전 확인
print("Pytorch 버전:", torch.__version__)

# CUDA 사용 가능 여부 확인
print("CUDA 사용 가능 여부:", torch.cuda.is_available())

# 사용 가능한 GPU 개수 확인
print("사용 가능한 GPU 개수:", torch.cuda.device_count())

# 현재 사용 중인 디바이스 ID
print("현재 디바이스 ID:", torch.cuda.current_device())

# 현재 디바이스 이름
print("현재 디바이스 이름:", torch.cuda.get_device_name(torch.cuda.current_device()))

# 텐서를 GPU에 할당해보기
x = torch.tensor([1.0, 2.0, 3.0]).to('cuda')
print("GPU에 할당된 텐서:", x)

** google colab에서는 무조건 실행이 가능해야 함.

jupyterlab

[local setting 시]

GPU 확인

1. 작업관리자에서 사용상태 확인

   작업관리자 → 성능 탭

   (옆에 보면 사용률 없음 확인)

1. 디스플레이 설정에서

   디스플레이 설정 → 고급 디스플레이 설정

   NVIDIA 없음 확인

1. 장치관리자에서

   장치관리자 → 디스플레이 어댑터 → NVIDIA GeForce 4060 확인

   

[NVIDIA install]