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?