torch程序无法调用GPU的问题

131 阅读1分钟

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

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

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

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

比如cuda100对应torch1.3.0