[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.int()

if torch.cuda.is_available():
    print('GPU를 사용할 수 있음')
    tensor = tensor.to('cuda')

print(f'shape: {tensor.shape}')
print(f'dtype: {tensor.dtype}')
print(f'device: {tensor.device}')

# 결과값 => 
# GPU를 사용할 수 있음
# shape: torch.Size([4, 3])
# dtype: torch.int32
# device: cuda:0

 

※ 만약 맥북을 사용하면서 jupyter notebook을 이용하여 GPU를 사용하고 싶으면 아래의 블로그를 토대로 설정하시면 됩니다.

 

https://selfhiam.tistory.com/109

맥북 M1, M2, M2 Pro Pytorch GPU가속화 사용하기