跑实验的时候,发现有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 | PyTorchhttps://pytorch.org/get-started/previous-versions/