在適配pytorch版的faster-rcnn(https://github.com/jwyang/faster-rcnn.pytorch.git)項目時遇到了這個問題,項目代碼是用的
pytorch-1.0分支。
Traceback (most recent call last):
File "train_net.py", line 226, in <module>
im_data = im_data.cuda()
File "/usr/local/lib/python2.7/dist-packages/torch/cuda/__init__.py", line 192, in _lazy_init
_check_driver()
File "/usr/local/lib/python2.7/dist-packages/torch/cuda/__init__.py", line 111, in _check_driver
of the CUDA driver.""".format(str(torch._C._cuda_getDriverVersion())))
AssertionError:
The NVIDIA driver on your system is too old (found version 10000).
Please update your GPU driver by downloading and installing a new
version from the URL: http://www.nvidia.com/Download/index.aspx
Alternatively, go to: https://pytorch.org to install
a PyTorch version that has been compiled with your version
of the CUDA driver.
我的環境如下:
其實答案已經很明顯了,只是我一眼沒看出來。一開始真的以爲如提示所言可能是cuda驅動版本太老,後面轉眼又想了一下
也許是我裝的pytorch版本太新。So,將pytorch由1.3.0切換到1.0問題解決,just so easy!