SystemInfo
這是一個用來獲取系統資訊的工具。它可以幫助你獲取 CPU、記憶體、磁盤、網路等系統資訊。
get_package_versions
-
說明:獲取常用深度學習和數據分析相關套件的版本。包含 PyTorch、PyTorch Lightning、TensorFlow、Keras、NumPy、Pandas、Scikit-learn、OpenCV 等套件的版本資訊。
-
傳回值
- dict:包含已安裝套件版本資訊的字典。
-
範例
import docsaidkit as D
versions_info = D.get_package_versions()
print(versions_info)
# versions_info = {
# 'PyTorch Version': '1.9.0',
# 'PyTorch Lightning Version': '1.3.8',
# 'TensorFlow Version': '2.5.0',
# 'Keras Version': '2.4.3',
# 'NumPy Version': '1.19.5',
# 'Pandas Version': '1.1.5',
# 'Scikit-learn Version': '0.24.2',
# 'OpenCV Version': '4.5.2'
# }
get_gpu_cuda_versions
-
說明:獲取 GPU 和 CUDA 版本資訊。嘗試使用 PyTorch、TensorFlow 和 CuPy 等套件來獲取 CUDA 版本,並使用
nvidia-smi
命令來獲取 Nvidia 驅動程式版本。 -
傳回值
- dict:包含 CUDA 和 GPU 驅動程式版本資訊的字典。
-
範例
import docsaidkit as D
gpu_cuda_info = D.get_gpu_cuda_versions()
print(gpu_cuda_info)
# gpu_cuda_info = {
# 'CUDA Version': '11.1',
# 'NVIDIA Driver Version': '460.32.03'
# }