I am having the same error:
Traceback (most recent call last):
  File "train_mtl.py", line 472, in <module>
    main()
  File "train_mtl.py", line 463, in main
    train(epoch)
  File "train_mtl.py", line 206, in train
    inputsBGR = Variable(inputsBGR.float().cuda())
RuntimeError: cuda runtime error (4) : unspecified launch failure at c:\new-builder_3\win-wheel\pytorch\aten\src\thc\generic/THCTensorCopy.cpp:20

I am running PyTorch 0.4.1 on Windows 10 with NVIDIA Quadro P3200 with Max-Q Design.
The algorithm that I am trying to train is retrieved from the following repository:

  
      
      GitHub
  
  
    

anilbatra2185/road_connectivity 1

Improved Road Connectivity by Joint Learning of Orientation and Segmentation (CVPR2019) - anilbatra2185/road_connectivity


  
  
    
    
  
  


My batch contains about 2000 images.