torch와 torch_cuda

torch

• 텐서 연산, 신경망 모듈, autograd functionality for automatic differentiation, 최적화 알고리즘 및 기타 유틸리티를 포함하는 핵심 PyTorch 라이브러리를 가져오게 됨.
• 이것은 기본 PyTorch 패키지이며, 명시적으로 지정하지 않는 한 GPU 연산을 명시적으로 처리하지 않음.

torch.cuda

• torch.cuda를 임포트하면 NVIDIA의 병렬 컴퓨팅 플랫폼이자 애플리케이션 프로그래밍 인터페이스(API) 모델인 CUDA(컴퓨팅 통합 장치 아키텍처)와 관련된 기능에 특별히 액세스하게 됨.
• PyTorch는 CUDA를 활용하여 고성능 수치 연산을 위해 NVIDIA GPU를 활용함.
• torch.cuda는 CUDA 장치 관리 및 상호 작용, GPU 메모리 할당, CUDA 커널 실행을 위한 함수와 유틸리티를 제공함.