确认驱动没问题(nvidia-smi 可以正常使用) 解决办法参照:https://github.com/pytorch/pytorch/issues/85773#issuecomment-1288033297
nvidia-smi
内容如下: