- 博客(2)
- 收藏
- 关注
转载 E tensorflow/stream_executor/cuda/cuda_dnn.cc:371] could not create cudnn handle: CUDNN_STATUS_INTE
tensorflow调用GPU时报错:再模型调用前加上如下代码:config = tf.ConfigProto(gpu_options=tf.GPUOptions(allow_growth=True))sess = tf.Session(config=config)自己的代码是:config = tf.ConfigProto(allow_soft_placement=True)sess...
2019-03-05 13:39:18 3489
原创 Input type (torch.cuda.FloatTensor) and weight type (torch.FloatTensor) should be the same
RuntimeError: Input type (torch.cuda.FloatTensor) and weight type (torch.FloatTensor) should be the same主要是由于训练时采用了cuda对自己定义的网络进行训练,但是后来调整网络又在自定义网络A外部添加了一层: def conv2x2(in_channel,out_channel): ...
2019-01-11 17:28:18 12128 2
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人