Pytorch问题踩坑总结：正确安装了GPU版本的pytorch，但是运行的时候总是用CPU

一直都以为是GPU版本的pytorch没安装成功

原来是因为设置了GPU id

查了一下，是因为设置了os.environ[“CUDA_VISIBLE_DEVICES”]=“1”之后，程序只能看到编码为1的GPU了,删除os.environ[“CUDA_VISIBLE_DEVICES”]=“1”就行了。