MS에서 제공하는 Simple BaseLine을 실행시키려는데
가상환경 셋팅에 굉장히 많은 시간과 애를 먹었습니다..
저는 CUDA 11.2 버전 + GTX 1080 Ti 인데
해당 스펙에 맞는 버전을 찾아서 해결했습니다.
pip install torch==1.7.1+cu110 torchvision==0.8.2+cu110 -f https://download.pytorch.org/whl/torch_stable.html
명령어를 보면 CUDA 11.0 버전으로 설치한 것을 알 수 있는데 11.2버전과도 호환 가능한 것으로 확인됩니다.
'머신러닝 & 딥러닝' 카테고리의 다른 글
| [openpose] openpose window demo 버전 실행하기(keypoints.json 얻기) (4) | 2023.05.18 |
|---|---|
| [openpose] 이미지에서 관절 포인트 검출하기 (2) | 2023.05.17 |