Question

这里是我正在使用的模型：

include-system-site-packages = true

这是我遇到的错误：

#import tensorflow as tf
def create_model():
    return tf.keras.models.Sequential([
    #tf.keras.layers.Flatten(input_shape=(2,)),
    tf.keras.layers.Conv2D(filters=32,kernel_size=(3,3),strides=(1,1),input_shape=(156,256,3),padding='valid',data_format='channels_last',
              activation='relu',kernel_initializer=tf.keras.initializers.he_normal(seed=0),name='Conv1'),
    tf.keras.layers.MaxPool2D(pool_size=(2,2),strides=(2,2),padding='valid',data_format='channels_last',name='Pool1'), 
    tf.keras.layers.Conv2D(filters=64,kernel_size=(3,3),strides=(1,1),padding='valid',data_format='channels_last',
              activation='relu',kernel_initializer=tf.keras.initializers.he_normal(seed=3),name='Conv2'),

    tf.keras.layers.Conv2D(filters=64,kernel_size=(3,3),strides=(2,2),padding='valid',data_format='channels_last',
              activation='relu',kernel_initializer=tf.keras.initializers.he_normal(seed=5),name='Conv3'),
    tf.keras.layers.MaxPool2D(pool_size=(2,2),strides=(1,1),padding='valid',data_format='channels_last',name='Pool2'),
    tf.keras.layers.Conv2D(filters=128,kernel_size=(3,3),strides=(2,2),padding='valid',data_format='channels_last',
              activation='relu',kernel_initializer=tf.keras.initializers.he_normal(seed=9),name='Conv4'),

    tf.keras.layers.MaxPool2D(pool_size=(2,2),strides=(2,2),padding='valid',data_format='channels_last',name='Pool3'),
    tf.keras.layers.Flatten(data_format='channels_last',name='Flatten'),    
    tf.keras.layers.Dense(units=30,activation='relu',kernel_initializer=tf.keras.initializers.glorot_normal(seed=32),name='FC1'),
    tf.keras.layers.Dense(units=15,activation='relu',kernel_initializer=tf.keras.initializers.glorot_normal(seed=33),name='FC2'),
    tf.keras.layers.Dense(units=8,activation='softmax',kernel_initializer=tf.keras.initializers.glorot_normal(seed=3),name='Output'),


  ])

我正在使用随anaconda cuda版本10.2安装的tensorflow 2.0。当我不使用cnn时，有人可以帮助我进行同样的安装吗？

是因为我正在使用CONV2d还是因为我正在使用Generator？我在装有16 GB RAM和4GB Nvidia 1650显卡的Windows 10计算机上。

Answer 1

遇到相同的错误，并通过以下方法解决：

gpus = tf.config.experimental.list_physical_devices('GPU')
tf.config.experimental.set_virtual_device_configuration(gpus[0], 
           [tf.config.experimental.VirtualDeviceConfiguration(memory_limit=4000)])

（带有GTX 1660，6G内存）

无法创建Cudnn句柄：CUDNN_STATUS_ALLOC_FAILED

1 个答案: