AbnerHqC / GaitSet

A flexible, effective and fast cross-view gait recognition network
582 stars 170 forks source link

AssertionError: Found no NVIDIA driver on your system. Please check that you have an NVIDIA GPU and installed a driver from http://www.nvidia.com/Download/index.aspx #95

Open sunny-lhm opened 4 years ago

sunny-lhm commented 4 years ago

AssertionError: Found no NVIDIA driver on your system. Please check that you have an NVIDIA GPU and installed a driver from http://www.nvidia.com/Download/index.aspx 我有GPU 和NVIDIA驱动但是为什么还是报错没有呀,难道是GPU型号不对我的GPU是Tesla k80的 Initialzing... Initializing data source... Data initialization complete. Initializing model... Traceback (most recent call last): File "test.py", line 35, in m = initialization(conf, test=opt.cache)[0] File "/home/xuhuanhuan_sunxuemei_cs/new/GaitSet-master/model/initialization.py", line 58, in initialization return initialize_model(config, train_source, test_source) File "/home/xuhuanhuan_sunxuemei_cs/new/GaitSet-master/model/initialization.py", line 47, in initialize_model m = Model(**model_param) File "/home/xuhuanhuan_sunxuemei_cs/new/GaitSet-master/model/model.py", line 61, in init self.encoder.cuda() File "/miniconda/envs/py36/lib/python3.6/site-packages/torch/nn/modules/module.py", line 260, in cuda return self._apply(lambda t: t.cuda(device)) File "/miniconda/envs/py36/lib/python3.6/site-packages/torch/nn/modules/module.py", line 187, in _apply module._apply(fn) File "/miniconda/envs/py36/lib/python3.6/site-packages/torch/nn/modules/module.py", line 187, in _apply module._apply(fn) File "/miniconda/envs/py36/lib/python3.6/site-packages/torch/nn/modules/module.py", line 187, in _apply module._apply(fn) [Previous line repeated 1 more time] File "/miniconda/envs/py36/lib/python3.6/site-packages/torch/nn/modules/module.py", line 193, in _apply param.data = fn(param.data) File "/miniconda/envs/py36/lib/python3.6/site-packages/torch/nn/modules/module.py", line 260, in return self._apply(lambda t: t.cuda(device)) File "/miniconda/envs/py36/lib/python3.6/site-packages/torch/cuda/init.py", line 161, in _lazy_init _check_driver() File "/miniconda/envs/py36/lib/python3.6/site-packages/torch/cuda/init.py", line 82, in _check_driver http://www.nvidia.com/Download/index.aspx""") AssertionError: Found no NVIDIA driver on your system. Please check that you have an NVIDIA GPU and installed a driver from http://www.nvidia.com/Download/index.aspx