.format(str(torch._C._cuda_getDriverVersion()))) AssertionError:

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.

 

把torch的版本降一下就好了，torchvision要对应（1.5->1.2,0.6->04.）