0

我需要一些帮助来为 scaled-yolov4 设置我的 linux 机器。我的 linux 盒子有一个 gpu 卡，Nvidia quadro P620 2GB 卡。运行 linux mint 20.4 = ubuntu 20.4。

我有蟒蛇设置。但没有太大帮助。这就是我所做的。

  1. 我使用以下命令安装了 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 没有内存不足错误。

  1. 我安装了 CUDA10.2，没有 GPU 也能正常工作，所以速度很慢。

有人能告诉我安装 NVIDIA 卡和 CUDA 并为 scaled-yolov4 设置环境的正确方法吗？非常感谢。

4

0 回答 0