开发验证

python3 - <<'PY'
import transformers 
import torch 
from transformers import BertConfig, BertModel  

print("transformers_version=" + transformers.__version__) 
print("torch_version=" + torch.__version__) 
print("torch_cuda=" + str(torch.version.cuda)) 
print("cuda_built=" + str(torch.backends.cuda.is_built())) 
print("cuda_available=" + str(torch.cuda.is_available())) 
config = BertConfig(hidden_size=32, num_hidden_layers=1, num_attention_heads=4, intermediate_size=64) 
model = BertModel(config) 
print("transformers_model_type=" + config.model_type) 
assert transformers.__version__ == "5.4.0"
assert config.model_type == "bert"
assert torch.backends.cuda.is_built() 
assert torch.cuda.is_available() 
PY

transformers_version=5.4.0 
torch_cuda=13.0 
cuda_built=True 
cuda_available=True 
transformers_model_type=bert