我已经尝试了几个小时来为 tensorflow-gpu 安装 CUDA。我觉得它已经启动并运行了,但在安装一个简单的 CNN 时出现了错误。我检查了我的 GPU,因为它似乎是内存不足错误。
nvidia-smi
返回:
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 495.29.05 Driver Version: 495.29.05 CUDA Version: 11.5 |
|-------------------------------+----------------------+----------------------+
| 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:01:00.0 On | N/A |
| N/A 38C P8 12W / N/A | 169MiB / 5938MiB | 3% Default |
| | | N/A |
+-------------------------------+----------------------+----------------------+
我在 ubuntu 20.04 LTS 上,使用最新的 nvidia-driver-495 和 RTX3060。`
我的装备:
sudo lshw -C display
返回:
*-display
description: VGA compatible controller
produit: NVIDIA Corporation
fabricant: NVIDIA Corporation
identifiant matériel: 0
information bus: pci@0000:01:00.0
version: a1
bits: 64 bits
horloge: 33MHz
fonctionnalités: pm msi pciexpress vga_controller bus_master cap_list rom
configuration : driver=nvidia latency=0
ressources : mémoireE/S:fa0-f9f mémoireE/S:fc0-fbf irq:95 mémoire:d0000000-d0ffffff mémoire:fa00000000-fbffffffff mémoire:fc00000000-fc01ffffff portE/S:2000(taille=128) mémoire:d1080000-d10fffff
这告诉我我的 GPU 有 6G 的 RAM。但我的 RTX 实际上有 16G。
有谁知道这可能来自哪里?
谢谢!