初始化网络时GPU内存不足

Question

我正在尝试初始化一个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)

Answer 1

当GPU驱动程序过时时，这个问题就发生了。 我的GPU是1070 4演出。 我建议重新安装驱动程序并重新启动。