머신러닝/2022-07/멜버른/ 파이토치 욜로5버전 개발환경 설정

 

2022년7월 현재 ... 파이썬 버전 3.9는 Fail 된다. 3.8 만 가능하다.

 

conda create -n yoloenv python=3.8

conda activate yoloenv

git clone https://github.com/ultralytics/yolov5

 

 conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=11.0 -c pytorch

 

10여분정도 소요된다. (PC사용에 따른 변동있음)

 

 

 

 

 

 

cuda 설치가 안되면 cuda:0 이 아니라  cpu 를 출력한다.

 

 

만약 cpu 로 출력되면, 아래와 같이 해결한다. (feat. 홍정명)

 

다음 경로로 이동한다.

C:\Users\[ 사용자 ]\AppData\Roaming\Python\Python39\site-packages

이곳에서 torch, torch-버전 폴더 두개 다 지운다.

 

그 다음 https://pytorch.org/get-started/locally/ 로 이동한다.

 

 

 

다시 설치하면 파이토치 GPU (CUDA 프로그래밍) 개발환경이 설정된다.