几种查看PyTorch、cuda 和 Python 版本方法小结
更新时间：2025年04月25日 08:32:44 作者：王红臣同学
本文主要介绍了几种查看PyTorch、cuda 和 Python 版本方法小结,除了直接使用 torch.__version__ 和 sys.version,我们还可以通过其他方式实现相同的功能,下面就一起来了解一下
在检查 PyTorch、cuda 和 Python 版本时，除了直接使用 torch.__version__ 和 sys.version，我们还可以通过其他方式实现相同的功能
方法 1：直接访问属性（原始代码）
import torch
import sys
print("PyTorch Version: {}".format(torch.__version__))
print("Python Version: {}".format(sys.version))
特点：
- 简单直接，无需额外依赖。
- 适用于快速检查版本信息。
方法 2：通过命令行工具
如果希望在脚本外部检查版本，可以直接使用命令行工具。
Python 版本
python --version # 或 python -V
PyTorch 版本
python -c "import torch; print(torch.__version__)"
特点：
- 适用于脚本外部检查，无需编写 Python 代码。
- 可以集成到 CI/CD 流程中。
方法 3：使用 torch.version 模块
PyTorch 提供了一个 torch.version 模块，可以获取更详细的版本信息。
import torch
import sys
# 获取 PyTorch 版本信息
print("PyTorch Version: {}".format(torch.version.__version__)) # 或直接使用 torch.__version__
print("PyTorch CUDA Version: {}".format(torch.version.cuda)) # 获取 CUDA 版本
print("PyTorch cuDNN Version: {}".format(torch.backends.cudnn.version())) # 获取 cuDNN 版本
# Python 版本
print("Python Version: {}".format(sys.version))
特点：
- 可以获取 CUDA 和 cuDNN 的版本信息，对于调试 GPU 环境非常有用。
torch.version提供了更细粒度的版本控制。
方法 4：使用 pkg_resources
pkg_resources 是 setuptools 提供的一个工具，可以查询已安装包的版本信息。
import pkg_resources
# 获取 PyTorch 版本
try:
pytorch_version = pkg_resources.get_distribution("torch").version
print("PyTorch Version: {}".format(pytorch_version))
except pkg_resources.DistributionNotFound:
print("PyTorch is not installed.")
# Python 版本仍通过 sys 模块
import sys
print("Python Version: {}".format(sys.version))
特点：
- 可以查询任何已安装包的版本，而不仅仅是 PyTorch。
- 如果包未安装，会捕获
DistributionNotFound异常。
方法 5：使用 platform 模块（补充 Python 信息）
虽然 sys.version 已经提供了 Python 版本信息，但 platform 模块可以提供更详细的系统信息。
import torch
import platform
print("PyTorch Version: {}".format(torch.__version__))
print("Python Version: {}".format(platform.python_version()))
print("Platform: {}".format(platform.platform()))
特点：
platform.platform()提供了操作系统的详细信息。- 适用于需要记录系统环境的场景。
方法 6：结合 subprocess 调用命令行
如果需要在 Python 脚本中调用外部命令行工具，可以使用 subprocess 模块。
import subprocess
def get_python_version():
result = subprocess.run(["python", "--version"], capture_output=True, text=True)
return result.stdout.strip()
def get_pytorch_version():
result = subprocess.run(["python", "-c", "import torch; print(torch.__version__)"], capture_output=True, text=True)
return result.stdout.strip()
print("Python Version: {}".format(get_python_version()))
print("PyTorch Version: {}".format(get_pytorch_version()))
特点：
- 适用于需要从外部命令行获取信息的场景。
- 可以灵活地调用其他命令行工具。
方法 7：使用 torch.utils.collect_env
PyTorch 提供了一个 torch.utils.collect_env 工具，可以收集详细的系统环境信息，包括 PyTorch、Python、CUDA、cuDNN 等。
import torch env_info = torch.utils.collect_env() print(env_info)
特点：
- 提供全面的环境信息，适合用于调试和问题报告。
- 输出格式为字典，可以进一步处理。
总结
| 方法 | 优点 | 缺点 |
|---|---|---|
| 直接访问属性 | 简单直接，无需额外依赖 | 功能有限，仅能获取基本版本信息 |
| PyTorch | ||
| 通过命令行工具 | 适用于脚本外部检查，无需编写 Python 代码 | 需要手动执行命令 |
使用 torch.version | 提供更详细的版本信息（CUDA、cuDNN） | 仅适用于 |
使用 pkg_resources | 可以查询任何已安装包的版本 | 需要额外依赖 setuptools |
使用 platform 模块 | 提供详细的系统信息 | 功能与 sys 模块部分重叠 |
结合 subprocess | 灵活调用外部命令行工具 | 实现复杂，性能可能较低 |
使用 torch.utils.collect_env | 提供全面的环境信息，适合调试 | 输出格式复杂，需要进一步处理 |
到此这篇关于几种查看PyTorch、cuda 和 Python 版本方法小结的文章就介绍到这了,更多相关PyTorch、cuda 和 Python 查看版本内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！
相关文章
这篇文章主要介绍了python闭包及装饰器运行原理解析,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下2020-06-06
这篇文章主要介绍了Linux下用Python脚本监控目录变化代码分享,本文直接给出实现代码,需要的朋友可以参考下2015-05-05
这篇文章主要介绍了Python实现PS滤镜碎片特效功能,结合实例形式分析了Python实现PS滤镜碎片效果的具体步骤与相关操作技巧,需要的朋友可以参考下2018-01-01
这篇文章主要介绍了python之关于数组和列表的区别及说明，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教2023-05-05
这篇文章主要和大家分享一个Python全自动输入文本的脚本，可以实现自动用Notepad++打开文本文件，然后自动输入文本，最后保存并关闭文件，从而实现全面自动化处理文本，希望对大家有所帮助2022-11-11
本文将介绍如何使用Python解析Laravel Cookie，以便在Web开发中处理这些Cookie数据，我们将深入了解Cookie的结构，以及如何在Python中对其进行解析和操作2023-12-12
登录验证码是每个网站登录时的基本标配，这篇文章主要介绍了Django实现登录随机验证码的示例代码，小编觉得挺不错的，现在分享给大家，也给大家做个参考。一起跟随小编过来看看吧2018-06-06
这篇文章主要为大家详细介绍了python实现多人聊天室，文中示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下2018-08-08
这章我们来介绍Python的扩展名之ctypes，教大家认识ctypes，有需要的朋友可以借鉴参考下，希望可以有所帮助，祝大家多多进步，早日升职加薪2021-09-09
logging是Python标准库中记录常用的记录日志库，主要用于输出运行日志，可以设置输出日志的等级、日志保存路径、日志文件回滚等。本文主要来和大家聊聊它的具体用法，希望对大家有所帮助2023-02-02
最新评论