CUDA는 사용하는게 까다롭기 보다는 환경 세팅하는 것이 제일 문제.
용량부족으로 SD카드 마운팅하다 잿자가 죽어버려서 cudnn까지 설치를 완료하지 못했음.
1.
cuda는 ubuntu 버전과의 호환이 중요 → seeed jetson xavier cuda 검색하면 seeed 페이지에서 cuda 11.0.2 버전을 설치하는 명령어를 정리해놨음
2.
일단 https://pytorch.kr/get-started/previous-versions/ 여기 피셜 cuda 11.0.2와 호환되는 토치는
pip install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2
일단 아슬아슬하게 YOLOv5의 최소 버전 조건에는 맞음
3.
그리고 검색한 곳으로 가서 해당 명령어 하란대로 따라서 설치하기
5.
여기서 압축파일을 불러오게 되는데
cd /home/download
tar xvzf ???.tgz
sudo cp cuda/include/cudnn* /usr/local/cuda-11.0/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda-11.0/lib64
sudo chmod a+r /usr/local/cuda-11.0/include/cudnn.h /usr/local/cuda-11.0/lib64/libcudnn*
XML
복사
cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2
XML
복사
요 명령어로 설치됐나 확인하기.