Check failed: CUDA_SUCCESS == dynload::cuCtxSetCurrent(cuda_context->context()) (0 vs. 4)

fullfanta / multimodal_transfer

tensorflow implementation of 'Multimodal Transfer: A Hierarchical Deep Convolutional Neural Network for Fast Artistic Style Transfer'

34 stars 9 forks source link

Check failed: CUDA_SUCCESS == dynload::cuCtxSetCurrent(cuda_context->context()) (0 vs. 4) #2

Open GongXinyuu opened 6 years ago

GongXinyuu commented 6 years ago

Hi, thx for your great code! However, when I run train.py, I've got Check failed: CUDA_SUCCESS == dynload::cuCtxSetCurrent(cuda_context->context()) (0 vs. 4)

Could you help me?