-1

私は本当にあなたの助けを求めています。

私は8vramであるGTX-1070を持っています。cuda 9.0 用に tensorflow-gpu、cuda 9.0、cudnn 7.0 をダウンロードしました。

すべてがDNNで正常に動作します。GPUも問題なく動いています。

しかし、画像に関係するモデルをトレーニングしようとすると、クラッシュします。現在、私は keras の事前トレーニング済み VGG16 を使用しています。

バッチサイズが小さく、サイズが64x64に縮小された画像を使用してみました。プロセスを見ると、GPU の使用率が 0% で、その後 100% まで急増してからクラッシュします。Spyder は、「カーネルが停止しました。再起動しています」と言います。

gtx-1070 は本当にメモリ不足ですか、それとも何か不足していますか?

読んでくれてありがとう

4

2 に答える 2

1

私が最初に試みることは Cudnn 7.1 をダウンロードすることです。

これらは従うべき適切な手順であり、Cuda 9 を再インストールすることを検討してください。私はある時点で同じことをしなければならなかった.

インストール手順

于 2018-04-13T19:14:24.933 に答える