我有 60000 张图像分为两类。我尝试使用预训练的 ResNet50 构建迁移学习，但我的新 GTX 1080 ti 在几个时期后返回 -1。我的猜测是内存不足。这很奇怪，因为使用相同的脚本和相同的数据，我可以在 GTX 1050 ti 上成功运行它。一切都在 tensorflow-gpu 后端
我已经在 ubuntu 上为 GTX 1080 ti 安装了 CUDA 9.0 和 cudnn 7.1.4。这是nvidia-smi
我看到缺少内存：
GTX 1080t ti 内存不足
数据挖掘
深度学习
分类
计算机视觉
迁移学习
2022-03-04 05:51:44
0个回答
没有发现任何回复~
其它你可能感兴趣的问题