collect_env()作用
时间: 2024-05-03 16:17:29 浏览: 11
`collect_env()`是PyTorch内部的一个函数,其作用是收集当前系统的环境信息,包括Python版本、PyTorch版本、CUDA版本、cuDNN版本、CPU信息和GPU信息等。该函数通常用于调试和排查问题,以便更好地理解系统的配置和性能。
`collect_env()`函数返回一个字典,其中包含了当前系统的环境信息,例如:
```
{
"sys.platform": "darwin",
"Python": "3.7.6 (default, Dec 19 2019, 23:50:31) \n[Clang 11.0.0 (clang-1100.0.33.16)]",
"PyTorch": "1.6.0",
"torchvision": "0.7.0",
"CUDA available": True,
"CUDA_HOME": "/usr/local/cuda",
"NVCC": "Cuda compilation tools, release 10.1, V10.1.243",
"GPU models": "GeForce GTX 1080 Ti",
"nvidia-smi": [
"Fri Oct 23 14:06:58 2020 ",
"+-----------------------------------------------------------------------------+",
"| NVIDIA-SMI 440.100 Driver Version: 440.100 CUDA Version: 10.2 |",
"|-------------------------------+----------------------+----------------------+",
"| GPU Name Persistence-M| Bus-Id Disp.A | Volatile Uncorr. ECC |",
"| Fan Temp Perf Pwr:Usage/Cap| Memory-Usage | GPU-Util Compute M. |",
"|===============================+======================+======================|",
"| 0 GeForce GTX 108... On | 00000000:02:00.0 Off | N/A |",
"| 27% 51C P2 61W / 250W | 850MiB / 11175MiB | 3% Default |",
"+-------------------------------+----------------------+----------------------+"
],
"CUDNN": "7.6.5.32",
"Device count": 1,
"Devices": [
{
"name": "GeForce GTX 1080 Ti",
"capability": "6.1",
"total_memory": 11175
}
],
"MAGMA": True,
"BLAS": "OpenBLAS",
"OMP": True
}
```
通过调用`collect_env()`函数,可以方便地查看当前系统的PyTorch和CUDA版本,以及GPU设备的信息等,从而更好地调试和优化PyTorch程序。