[GoogleColab] 코렙에서 GPU 사용방법
코렙의 GPU 사용 1. 코렙에서 device 변경하는 방법 상단 메뉴 -> 런타임 -> 런타임 유형 변경 -> 하드웨어 가속기를 T4 GPU or TPU로 변경 -> 저장 -> 세션 다시 시작 및 모두 실행 tensor = torch.rand(3, 4) print(f'shape: {tensor.shape}') print(f'dtype: {tensor.dtype}') print(f'device: {tensor.device}') # 결과값 => # shape: torch.Size([3, 4]) # dtype: torch.float32 # device: cpu # is_available() : gpu 사용할 수 있는지 여부 tensor = tensor.reshape(4, 3) tensor = tensor.i..
2024.01.09