cuda和pytorch的对应版本问题

跑实验的时候,发现有cuda和pytorch不匹配的情况,报错如下所示:

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  ../c10/cuda/CUDAFunctions.cpp:112.)

解决方法很简单,直接在下面找对应的语句即可。

Previous PyTorch Versions | PyTorchicon-default.png?t=M276https://pytorch.org/get-started/previous-versions/

你可能感兴趣的:(tricks,深度学习)