RuntimeError: cuda runtime error (8) : invalid device function

[limusen123]

My CUDA is 10.0, Python is 3.6, GPU computing power is 5.0, tensorflow version is 1.14.0, running train.py is wrong. Runtime Error: CUDA runtime error (8): invalid device function Another thing I don't understand is, when I run train. py, what is the number of data I read in 261, and where does 261 come from? 0%| | 0/261 [00:00<?,? It/s]

[leftthomas]

@limusen123

1. 261 is computed by DataLoader according to the batch_size and Len(dataset)
2. CUDA runtime error is caused by pertained model, you should load model to cpu and then convert it to cuda