728x90 반응형 파이토치 gpu1 [GPU 환경 세팅]GPU와 파이토치 with Window [교훈] gpu 툴킷과 파이토치 버전을 필수적으로 확인하라~!! 기본적으로 리눅스와 친한 친구들이기 때문에 window와 호환이 잘되는 친구들을 적절히 찾아서 사용해야 한다. gpu툴킷과 호환되는 파이토치의 버전이 존재한다. 텐서2.0버전 이상은 기본으로 gpu를 활용한다. gpu의 성능에 따라 배치값을 조절 or yolov5폴더에 있는 util폴더 → datasets.py에서 nw를 고정해줘야함 2021년 11월 16일 참고 사이트: [Windows 10] Pytorch GPU 사용을 위한 CUDA 환경 만들기 Pytorch 사용을 위한 Cuda 환경구축에 대한 글입니다. Anaconda3를 이미 설치한 상태에서 설치를 진행했습니다. [GPU 환경 세팅] 1. 그래픽 드라이버 설치 우선 자신이 가지고 .. 2023. 4. 26. 이전 1 다음 728x90 반응형