Как настроить среду для Scaled-YOLOv4 в Linux
Мне нужна помощь в настройке моего Linux-бокса для scaled-yolov4. В моем Linux-боксе есть карта gpu, карта Nvidia quadro P620 2GB. запущен linux mint 20.4 = ubuntu 20.4.
У меня есть установка анаконды. Но не сильно помогло. Это то, что я сделал.
- Я установил cuda 11.2, используя следующее:
sudo mv cuda-ubuntu2004.pin /etc/apt/preferences.d/cuda-repository-pin-600
wget https://developer.download.nvidia.com/compute/cuda/11.2.1/local_installers/cuda-repo-ubuntu2004-11-2-local_11.2.1-460.32.03-1_amd64.deb
sudo dpkg -i cuda-repo-ubuntu2004-11-2-local_11.2.1-460.32.03-1_amd64.deb
sudo apt-key add /var/cuda-repo-ubuntu2004-11-2-local/7fa2af80.pub
sudo apt-get update
sudo apt-get -y install cuda
а затем с помощью conda установить остальные. В этом случае detect.py работает нормально, но train.py не работал с ошибкой нехватки памяти.
- Я установил CUDA10.2, все работает без графического процессора, поэтому он очень медленный.
Может кто-нибудь подскажет, как правильно установить карту NVIDIA и CUDA и настроить среду для scaled-yolov4? Большое спасибо.