ubuntu - RTX 3060 无法使用 cuda 11.4 运行 Pytorch Yolov4

Question

在我使用 RTX2070 SUPER 运行 Pytorch Yolov4 之前，现在我的 PC 已更改为使用 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

Answer 1

通过在我的 Conda Env 中重新安装 pytorch 来解决。
您可以尝试重新安装 Pytorch 或创建一个新的 Conda 环境以再次执行此操作。