Giters

leftthomas / ESPCN

A PyTorch implementation of ESPCN based on CVPR 2016 paper "Real-Time Single Image and Video Super-Resolution Using an Efficient Sub-Pixel Convolutional Neural Network"

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

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

limusen123 opened this issue · comments

commented

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]

commented

@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