我正在尝试初始化CNN，然后将其放在我的GPU上进行训练。当我将其放在GPU上时，出现错误：（CUDA错误：内存不足）。我已经运行了类似的网络，但没有出现此类问题。这是cuda中唯一的事情，因为我尚未加载任何图像。关于出什么问题有什么想法吗？
我正在GTX 1070ti 8GB上使用pytorch版本0.4.1。
| NVIDIA-SMI 410.104 Driver Version: 410.104 CUDA Version: 10.0 |
|-------------------------------+----------------------+----------------------+
| GPU Name Persistence-M| Bus-Id Disp.A | Volatile Uncorr. ECC |
| Fan Temp Perf Pwr:Usage/Cap| Memory-Usage | GPU-Util Compute M. |
|===============================+======================+======================|
| 0 GeForce GTX 107... Off | 00000000:01:00.0 On | N/A |
| 0% 43C P2 39W / 180W | 8024MiB / 8111MiB | 0% Default |
+-------------------------------+----------------------+----------------------+
+-----------------------------------------------------------------------------+
| Processes: GPU Memory |
| GPU PID Type Process name Usage |
|=============================================================================|
| 0 1129 G /usr/lib/xorg/Xorg 36MiB |
| 0 1164 G /usr/bin/gnome-shell 57MiB |
| 0 1415 G /usr/lib/xorg/Xorg 200MiB |
| 0 1548 G /usr/bin/gnome-shell 90MiB |
| 0 6323 C /usr/bin/python3 525MiB |
| 0 9521 C /usr/bin/python3 1827MiB |
| 0 18821 C /usr/bin/python3 4883MiB |
| 0 27137 G ...uest-channel-token=16389326112703159917 45MiB |
| 0 29161 C /usr/bin/python3 355MiB |
我尝试减小线性层的大小，但是没有运气。
net = piccnn()
net.to(device)
答案 0 :(得分：1)
我曾经在GPU驱动程序过时时发生此问题。我的GPU是1070 4演出。我建议重新安装驱动程序并重新启动。