错误:UserWarning: CUDA initialization: The NVIDIA driver on your system is too old (found version 9010). 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:100.)
return torch._C._cuda_getDeviceCount() > 0
解决方法:
解决方法如下 https://www.it610.com/article/1282995195515781120.htm(ubuntu)
win查看nvidia-smi.exe:win下该程序位置为C:\Program Files\NVIDIACorporation\NVSMI\nvidia-smi.exe,或者手动搜索到该位置即可
cuda下载安装教程(通过conda):https://mp.weixin.qq.com/s/QConpJvQbtWRXCgdtaQhnQ