在适配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!

Logo

旨在为数千万中国开发者提供一个无缝且高效的云端环境,以支持学习、使用和贡献开源项目。

更多推荐