如何通过pytorch查询当前的CUDA版本？

本文共计311个文字，预计阅读时间需要2分钟。

由于PyTorch的whl安装包名称都一样，我们很难区分基于CUDA的哪个版本。可以使用以下命令查看安装的CUDA版本：`import torch; print(torch.version.cuda)`。此外，了解PyTorch的网络定义参数也很重要，因为有时网络参数设置不当会导致问题。例如，如果网络定义参数设置错误，可能会出现网络定义参数后面不能‘‘的错误。

由于pytorch的whl 安装包名字都一样，所以我们很难区分到底是基于cuda 的哪个版本。