torch程序无法调用GPU的问题

1、你的gpu序号是否写对了

nvidia-smi看看,一定是0、1等序号

torch程序无法调用GPU的问题_第1张图片

2、你的torch和cuda版本号是否匹配

打开下面网页看看,一定要匹配:https://download.pytorch.org/whl/torch_stable.html

比如cuda100对应torch1.3.0

torch程序无法调用GPU的问题_第2张图片

你可能感兴趣的:(机器学习)