微信公众号: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