错误原因:python中安装的pytorch对应的CUDA版本和服务器上的显卡驱动对应的CUDA版本不兼容,显卡驱动对应的CUDA版本应当>=pytorch对应的CUDA版本。
显卡驱动对应的CUDA版本获取:执行命令nvidia-smi
,右上角CUDA Version
对应的就是显卡驱动对应的CUDA版本。
然后到pytorch官网找对应的版本
根据服务器的CUDA版本选取合适的pytorch对应的CUDA版本,比如,我的服务器CUDA版本为11.0,所以我也要找pytorch对应的CUDA版本,然后再看看哪个pytorch版本适合你的项目。然后安装相应的版本即可。