아래 답변은 OpenAI의 GPT-4 모델로 자동 생성한 것으로, 정확하지 않을 수 있습니다.
잘못된 답변을 보셨다면 댓글로 첨삭 지도 부탁드립니다.
안녕하세요! PyTorch 사용자 모임에서 강의를 들으셨다니 기쁩니다. CUDA 설정 문제를 해결해 드리겠습니다.
먼저, PyTorch 버전을 확인해주세요. 최신 버전을 사용하는 것이 좋습니다. 아래 명령어로 설치된 PyTorch 버전을 확인할 수 있습니다.
import torch
print(torch.__version__)
만약 버전이 낮다면, 최신 버전으로 업그레이드해주세요.
pip install --upgrade torch torchvision
다음으로, PyTorch가 GPU를 사용할 수 있는지 확인해봅시다. 아래 코드를 실행해보세요.
import torch
print(torch.cuda.is_available())
만약 여전히 False가 출력된다면, 다음 단계를 따라해주세요.
NVIDIA GPU 드라이버를 최신 버전으로 업데이트합니다. NVIDIA 공식 웹사이트에서 드라이버를 다운로드하실 수 있습니다.
CUDA와 cuDNN을 설치합니다. PyTorch와 호환되는 CUDA 버전을 설치해야 합니다. PyTorch 1.10.0 기준으로 CUDA 11.3을 지원하므로, 이 버전을 설치하는 것이 좋습니다. CUDA를 설치한 후, cuDNN 라이브러리도 설치해주세요. CUDA와 cuDNN 설치 방법은 이 공식 문서를 참고하시면 됩니다.
환경 변수를 설정합니다. 시스템 환경 변수에 CUDA_HOME, CUDA_PATH, PATH를 추가해주세요. 아래와 같이 설정하시면 됩니다.