【发布时间】：2021-08-20 04:34:36
【问题描述】：
之前我用 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
【问题讨论】：
-
你是如何安装 PyTorch 的？
-
嗨@Berriel，很抱歉我的回复晚了。我通过在 Conda env 中重新安装 pytorch 解决了这个问题。谢谢