RTX3060 不能 运行 Pytorch Yolov4 with cuda11.4
RTX3060 cannot run Pytorch Yolov4 with cuda11.4
之前我使用 RTX2070 SUPER 到 运行 Pytorch Yolov4，现在我的电脑改用 RTX3060，ASUS KO GeForce RTX™ 3060 OC。
我已经删除了现有的cuda11.2并用cuda11.4和Nvidia Driver 470.57.02重新安装
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 470.57.02 Driver Version: 470.57.02 CUDA Version: 11.4 |
|-------------------------------+----------------------+----------------------+
| GPU Name Persistence-M| Bus-Id Disp.A | Volatile Uncorr. ECC |
| Fan Temp Perf Pwr:Usage/Cap| Memory-Usage | GPU-Util Compute M. |
| | | MIG M. |
|===============================+======================+======================|
| 0 NVIDIA GeForce ... On | 00000000:07:00.0 Off | N/A |
| 0% 42C P8 16W / 170W | 403MiB / 12053MiB | 0% Default |
| | | N/A |
+-------------------------------+----------------------+----------------------+
+-----------------------------------------------------------------------------+
| Processes: |
| GPU GI CI PID Type Process name GPU Memory |
| ID ID Usage |
|=============================================================================|
| 0 N/A N/A 1236 G /usr/lib/xorg/Xorg 9MiB |
| 0 N/A N/A 1264 G /usr/bin/gnome-shell 6MiB |
| 0 N/A N/A 2124 C python 153MiB |
+-----------------------------------------------------------------------------+
然而，对于 cuda11.4 和 RTX3060，我无法 运行 Pytorch Yolov4 检测。我运行检测的时候，加载权重后检测会卡住，Loading weights from ./data/people.weights... Done!。同时，nvidia-smi可以显示“python”（PID 2124以上）正在使用GPU显存，并且“python”使用的GPU显存会不断增加。
难道cuda11.4还不支持RTX3060或者Pytorch1.4吗？
Environment:
ASUS KO GeForce RTX™ 3060 OC
Ubuntu 18.04.5 LTS
cuda 11.4
nvidia driver 470.57.02
conda 4.8.3
python 3.8.5
pytorch 1.4
通过在我的 Conda 环境中重新安装 pytorch 解决
您可以尝试重新安装 Pytorch 或创建一个新的 Conda 环境来重新安装。
之前我使用 RTX2070 SUPER 到 运行 Pytorch Yolov4，现在我的电脑改用 RTX3060，ASUS KO GeForce RTX™ 3060 OC。
我已经删除了现有的cuda11.2并用cuda11.4和Nvidia Driver 470.57.02重新安装
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 470.57.02 Driver Version: 470.57.02 CUDA Version: 11.4 |
|-------------------------------+----------------------+----------------------+
| GPU Name Persistence-M| Bus-Id Disp.A | Volatile Uncorr. ECC |
| Fan Temp Perf Pwr:Usage/Cap| Memory-Usage | GPU-Util Compute M. |
| | | MIG M. |
|===============================+======================+======================|
| 0 NVIDIA GeForce ... On | 00000000:07:00.0 Off | N/A |
| 0% 42C P8 16W / 170W | 403MiB / 12053MiB | 0% Default |
| | | N/A |
+-------------------------------+----------------------+----------------------+
+-----------------------------------------------------------------------------+
| Processes: |
| GPU GI CI PID Type Process name GPU Memory |
| ID ID Usage |
|=============================================================================|
| 0 N/A N/A 1236 G /usr/lib/xorg/Xorg 9MiB |
| 0 N/A N/A 1264 G /usr/bin/gnome-shell 6MiB |
| 0 N/A N/A 2124 C python 153MiB |
+-----------------------------------------------------------------------------+
然而，对于 cuda11.4 和 RTX3060，我无法 运行 Pytorch Yolov4 检测。我运行检测的时候，加载权重后检测会卡住，Loading weights from ./data/people.weights... Done!。同时，nvidia-smi可以显示“python”（PID 2124以上）正在使用GPU显存，并且“python”使用的GPU显存会不断增加。
难道cuda11.4还不支持RTX3060或者Pytorch1.4吗？
Environment:
ASUS KO GeForce RTX™ 3060 OC
Ubuntu 18.04.5 LTS
cuda 11.4
nvidia driver 470.57.02
conda 4.8.3
python 3.8.5
pytorch 1.4
通过在我的 Conda 环境中重新安装 pytorch 解决
您可以尝试重新安装 Pytorch 或创建一个新的 Conda 环境来重新安装。