【Python】【Pytorch】【CUDA】之程序卡在.cuda()不出来

本人环境：

cuda 11.4

由于之前安装pytorch版本没有匹配，在此环境，将模型通过.cuda()往GPU上部署的时候，发现程序一直出不来。

很多人都遇到过类似问题，参考：CUDA版本11.4，pytorch应该下哪个版本的？ - 知乎

python - Pytorch get stuck at .to(‘cuda’) or .cuda() - Stack Overflowhttps://stackoverflow.com/questions/57566057/pytorch-get-stuck-at-to-cuda-or-cuda nn.DataParallel(model).cuda() stuck - PyTorch Forumshttps://discuss.pytorch.org/t/nn-dataparallel-model-cuda-stuck/8085

最后发现是由于pytorch版本跟cuda版本不匹配，pytorch官网最新支持cuda11.3

最后尝试将pytorch更新到最新版本11.3，问题得以解决。暂时先这样用着。