Because of some issue i removed pytorch and again installed it, after that i am getting this error on running my models
File "train_unet.py", line 46, in <module>
    g = UNet(3).type(dtype)
  File "/data/anaconda2/lib/python2.7/site-packages/torch/nn/modules/modul
e.py", line 277, in type
    return self._apply(lambda t: t.type(dst_type))
  File "/data/anaconda2/lib/python2.7/site-packages/torch/nn/modules/modul
e.py", line 185, in _apply
    module._apply(fn)
  File "/data/anaconda2/lib/python2.7/site-packages/torch/nn/modules/modu$
e.py", line 185, in _apply
    module._apply(fn)

  File "/data/anaconda2/lib/python2.7/site-packages/torch/nn/modules/modu$
e.py", line 185, in _apply
    module._apply(fn)
  File "/data/anaconda2/lib/python2.7/site-packages/torch/nn/modules/modu$
e.py", line 185, in _apply
    module._apply(fn)
  File "/data/anaconda2/lib/python2.7/site-packages/torch/nn/modules/modu$
e.py", line 191, in _apply
    param.data = fn(param.data)
  File "/data/anaconda2/lib/python2.7/site-packages/torch/nn/modules/modu$
e.py", line 277, in <lambda>
    return self._apply(lambda t: t.type(dst_type))
RuntimeError: CUDA error: unknown error

How to fix this?