【亲测解决】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

猜你喜欢

转载自blog.csdn.net/woai8339/article/details/140323841