CUDA1 [PyTorch] CUDA와 자동미분 이수안컴퓨터연구소의 파이토치(PyTorch) 기초 영상을 보고 정리한 내용입니다. ✅ CUDA Tensor CUDA(Computed Unified Device Architecture)는 NVIDIA의 GPU 개발 툴이다. 파이토치는 .to() 메서드를 통해 텐서를 cpu 또는 gpu 등 어떠한 장치로도 옮길 수 있다. import torch x = torch.randn(1) print(x) print(x.item()) print(x.dtype) # output # tensor([-0.3513]) # -0.35131844878196716 # torch.float32 스칼라 값을 가지는 텐서를 하나 생성한 뒤 텐서를 살펴보면 위와 같다. -0.3513... 이라는 값을 가지고 있고 데이터 타입은 float3.. 2022. 2. 23. 이전 1 다음
