Traceback (most recent call last):
File "C:/Users/Lenovo/Desktop/deeplabv3-plus-pytorch-main/predict.py", line 17, in
deeplab = DeeplabV3()
File "C:\Users\Lenovo\Desktop\deeplabv3-plus-pytorch-main\deeplab.py", line 85, in init
self.generate()
File "C:\Users\Lenovo\Desktop\deeplabv3-plus-pytorch-main\deeplab.py", line 105, in generate
self.net = self.net.cuda()
File "F:\anaconda\envs\pytorch3\lib\site-packages\torch\nn\modules\module.py", line 311, in cuda
return self._apply(lambda t: t.cuda(device))
File "F:\anaconda\envs\pytorch3\lib\site-packages\torch\nn\modules\module.py", line 208, in _apply
module._apply(fn)
File "F:\anaconda\envs\pytorch3\lib\site-packages\torch\nn\modules\module.py", line 208, in _apply
module._apply(fn)
File "F:\anaconda\envs\pytorch3\lib\site-packages\torch\nn\modules\module.py", line 208, in _apply
module._apply(fn)
File "F:\anaconda\envs\pytorch3\lib\site-packages\torch\nn\modules\module.py", line 230, in _apply
param_applied = fn(param)
File "F:\anaconda\envs\pytorch3\lib\site-packages\torch\nn\modules\module.py", line 311, in
return self.apply(lambda t: t.cuda(device))
File "F:\anaconda\envs\pytorch3\lib\site-packages\torch\cuda_init.py", line 178, in _lazy_init
check_driver()
File "F:\anaconda\envs\pytorch3\lib\site-packages\torch\cuda_init.py", line 92, in _check_driver
raise AssertionError("Torch not compiled with CUDA enabled")
AssertionError: Torch not compiled with CUDA enabled
cuda和cudnn安装好了没有？torch是否为GPU版本？