I installed cuda 11.2, cudnn 8.1 and tensorflow 2.8 by following the windows native installation guide. I was running training on a model and it was training fine with my GPU as intended. I then changed the model from a custom model to a sequential one with the following code
model = Sequential()
model.add(Conv2D(3, (3, 3), activation='relu', padding='same', input_shape=(540, 960, 2*3)))
model.compile(optimizer='adam', loss='mse', metrics=['accuracy'])
model.fit(x=numpy.array(x),
y=numpy.array(y), epochs=50, batch_size=16, validation_split=0.1)
model.save("model/interpolation_model2")
However after doing this code I’m getting the following error
Epoch 1/50
Could not locate cublasLt64_12.dll. Please make sure it is in your library path!