查看torch是否可以调用GPU、CUDA版本、cuDNN版本、torch版本、transformers版本，更新pip版本

1.查看GPU是否能够运行

import torch
print(torch.cuda.is_available())

True

2.查看CUDA版本和cuDNN版本

使用系统交互：

nvcc --version

nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2019 NVIDIA Corporation
Built on Wed_Oct_23_19:32:27_Pacific_Daylight_Time_2019
Cuda compilation tools, release 10.2, V10.2.89

下载了torch：

import torch

print(torch.version.cuda)
print(torch.backends.cudnn.version())

10.1
7604

 

3.查看torch版本和transformers版本 

import torch
import numpy as np
import transformers
print(torch.__version__)  # 1.7.1
print(transformers.__version__) # 2.1.1

1.7.1
3.4.0

4.更新pip版本 

pip install --upgrade pip