RuntimeError: cuDNN error: CUDNN_STATUS_INTERNAL_ERROR

  

   

    

     

    
    

     

      params = {
     
    
   

    

     

    
    

     

          "model": "UNet11",
     
    
   

    

     

    
    

     

          "device": "cuda",
     
    
   

    

     

    
    

     

          "lr": 0.001,
     
    
   

    

     

    
    

     

          "batch_size": 16,
     
    
   

    

     

    
    

     

          "num_workers": 4,
     
    
   

    

     

    
    

     

          "epochs": 10,
     
    
   

    

     

    
    

     

      }
     
    
   

    

     

    
    

     

      model = create_model(params)
     
    
  
 

将num_workers改为0

如果还是没有解决，试试将batchsize设置小一点。

文章来源: wanghao.blog.csdn.net，作者：AI浩，版权归原作者所有，如需转载，请联系作者。

原文链接：wanghao.blog.csdn.net/article/details/116667008