https://teang1995.tistory.com/17
[ 개발 환경 설정 ] Windows 10 에 CUDA를 설치해보자!
아! 입사 첫 주라 설정해줄 것이 정말 징글징글하게 많다! 전 게시물이 conda + vscode 였다면, 이번엔 CUDA + Torch다! conda 설정 및 python 개발 환경 구축 : link 1. 내 GPU 확인 어렵지 않다. 우선 시작 메뉴
teang1995.tistory.com
위 블로그의 과정 그대로 진행한다.
CUDA 설치 및 설치 실패 대처
목적 1) CUDA 설치 및 CUDA 설치 확인 2) CUDA 설치 오류 대처 1) CUDA 설치 및 CUDA 설치 확인 CUDA 11.0버전을 설치했습니다. Visual Studio에서 프로젝트에 CUDA 툴킷 종속 설정 2) CUDA 설치 오류 대처 CUDA설치 후
kjk92.tistory.com
오류가 나타난다면 위의 글을 참고한다.
nvcc -V 명령어로 cmd에서 내 CUDA 버전 확인
PyTorch
An open source machine learning framework that accelerates the path from research prototyping to production deployment.
pytorch.org
비슷해 보이는 12.1을 명령어를 복사하여 cmd창에 붙여넣어 설치한다.
이때 권한 문제로 설치가 안되는 오류가 나타나면 cmd를 관리자 권한으로 실행하여 해결 가능하다
명령어를 입력하여 설치하였다.
cmd창에서 python을 실행해 위의 명령어를 입력하거나 vscode에서 입력하면 True로 나온다
난 처음엔 False로 나왔었는데 이는 CUDA 버전은 12.x 인데 상술한 과정에서 torch를 설치할 때 실수로 CUDA 11.x 버전을 선택하고 설치해서 호환 오류가 생겼다.
그래픽 카드까지 정상적으로 출력된다.
'프로그래밍 > 데이터 사이언스 공부' 카테고리의 다른 글
| ByteTrack 패키지에서 import yolox 오류 (0) | 2023.10.19 |
|---|---|
| Yolov8s 모델에 바로 resume 옵션을 사용하는건 에러가 나온다 (0) | 2023.10.18 |
| yolov8를 mac m1칩으로 훈련 (0) | 2023.10.18 |
| 중단된 Yolov8 모델 훈련을 이어서할 때 필수인 resume 옵션 (1) | 2023.10.17 |
| YOLOv8 차량 추적, 집계 실습 (0) | 2023.09.23 |