2021-05-22 Please update your GPU driver by downloading and installing a new version from the URL: h

(pyenv3) [root@bsyocr server]# python nlp/run.py > nlp/log.txt
/opt/pyenv3/lib64/python3.6/site-packages/torch/cuda/__init__.py:52: UserWarning: CUDA initialization: The NVIDIA driver on your system is too old (found version 10010). 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. (Triggered internally at  /pytorch/c10/cuda/CUDAFunctions.cpp:109.)
  return torch._C._cuda_getDeviceCount() > 0
38185it [00:52, 734.24it/s]
4242it [00:05, 791.65it/s]
4707it [00:06, 781.75it/s]
/opt/server/pytorch_pretrained/optimization.py:275: UserWarning: This overload of add_ is deprecated:
    add_(Number alpha, Tensor other)
Consider using one of the following signatures instead:
    add_(Tensor other, *, Number alpha) (Triggered internally at  /pytorch/torch/csrc/utils/python_arg_parser.cpp:1005.)
  next_m.mul_(beta1).add_(1 - beta1, grad)

 

 

参考

1 解决思路(不盲目升级cuda,调整pytorch版本)

https://www.zhihu.com/question/309583980

 

2 cuda 与 pytorch对应关系

https://blog.csdn.net/kellyroslyn/article/details/109668001

 

你可能感兴趣的:(大数据)