Pytorch的一些使用技巧（不定期更新）

1.如果pytorch运行有问题可以查看一下运行环境以及编译的cuda版本等：

python -c 'from torch.utils.collect_env import main; main()'

示例输出：

Collecting environment information... PyTorch version: 1.4.0 
Is debug build: No 
CUDA used to build PyTorch: 10.1

OS: CentOS Linux 7 (Core) GCC version: (GCC) 4.8.5 20150623 (Red Hat
4.8.5-39) CMake version: Could not collect

Python version: 3.7 Is CUDA available: Yes 
CUDA runtime version: Could not collect 
GPU models and configuration:  
GPU 0: Tesla V100-PCIE-32GB 
GPU 1: Tesla V100-PCIE-32GB 
GPU 2: Tesla V100-PCIE-32GB 
GPU 3: Tesla V100-PCIE-32GB

Nvidia driver version: 418.87.00 cuDNN version: Could not collect

Versions of relevant libraries: 
[pip3] numpy==1.19.0 
[pip3] torch==1.4.0 
[pip3] torchvision==0.5.0 
[conda] torch                  1.4.0                    pypi_0    pypi 
[conda] torchvision               0.5.0                    pypi_0    pypi

2.快速测试pytorch运行情况

python -c 'import torch;print(torch.eye(3))'