인프런 커뮤니티 질문&답변
작성한 질문수
Pytorch 설치 GPU 버전으로 다운해도 되나요?
해결된 질문
작성
·
126
답변 2
1
안녕하세요, 인프런 AI 인턴입니다. 하준서님.
PyTorch를 사용할 때 GPU 버전을 설치하는 것은 매우 유익합니다. GPU는 딥러닝 모델 학습 속도를 크게 향상시킬 수 있기 때문입니다. 특히 RTX 4060과 같은 최신 GPU를 가지고 계시다면 GPU 버전의 PyTorch 설치를 권장합니다.
기본적으로 PyTorch는 다음과 같이 설치할 수 있습니다:
pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu117
여기서 cu117은 CUDA 11.7 버전의 PyTorch를 의미합니다. 설치 과정에서 문제가 발생할 수 있으므로, CUDA 및 CUDNN 버전이 컴퓨터에 올바르게 설치되어 있는지 확인해 주세요.
관련하여 다음 게시글들을 참조하시면 도움이 될 것입니다:
- 파이토치 device (gpu / cpu) 관련 질문입니다 - 텐서가 GPU 또는 CPU에 저장되는 방식을 설명한 답변입니다.
의견이 도움이 되셨기를 바랍니다. 추가적으로 궁금한 점이 있으시면, 이어서 질문해 주세요. 곧 지식공유자께서 답변해 주실 것입니다.
저는 질문자님의 이해를 돕고자 지식공유자님의 기존 답변을 참고하여 유사한 답변 링크를 추천해드리고 있습니다.
현재 베타 기능으로 답변이 만족스럽지 않을 수 있는 점 양해 부탁드립니다. 🙏
추가적으로 궁금한 점이 있으시면, 이어서 질문해 주세요. 곧 지식공유자께서 답변해 주실 것입니다.
1
