不能调用gpu,pytorch 与cuda不兼容

>>> import torch
>>> torch.__version__
'1.4.0'
>>> torch.zeros((2,3)).cuda()
Traceback (most recent call last):
  File "", line 1, in
  File "/data/wanglei/anaconda3/envs/point2mesh/lib/python3.8/site-packages/torch/cuda/__init__.py", line 196, in _lazy_init
    _check_driver()
  File "/data/wanglei/anaconda3/envs/point2mesh/lib/python3.8/site-packages/torch/cuda/__init__.py", line 104, in _check_driver
    raise AssertionError("""
AssertionError:
The NVIDIA driver on your system is too old (found version 9000).
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.
 

你可能感兴趣的:(不能调用gpu,pytorch 与cuda不兼容)