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!