【亲测解决】CUDA initialization error with `torch.zeros([1]).cuda()`

微信公众号：leetcode_algos_life，代码随想随记
小红书：412408155
CSDN：https://blog.csdn.net/woai8339?type=blog ，代码随想随记
GitHub: https://github.com/riverind
抖音【暂未开始，计划开始】：tian72530，代码随想随记
知乎【暂未开始，计划开始】：代码随想随记

【背景】
安装torch后，发现GPU被mig了。
nvidia-smi -L可以看到

【解决方案】
禁用GPU的MIG功能。

方法一：

nvidia-smi -mig 0

方法二：
如果方法一失效，

fuser -v /dev/nvidia*
pkill -u nobody【username】
nvidia-smi -mig 0

【核验】
输入 nvidia-smi -L 查看，只有GPU没有出现MIG就对了。

Ref，
https://discuss.pytorch.org/t/cuda-initialization-error-with-torch-zeros-1-cuda/183460