mirror of
https://github.com/taisuii/ClassificationCaptchaOcr.git
synced 2026-03-18 07:03:17 +08:00
3 lines
142 B
Python
3 lines
142 B
Python
import torch
|
|
print(torch.version.cuda) # 检查 PyTorch 是否包含 CUDA 支持
|
|
print(torch.cuda.is_available()) # 检查 CUDA 是否可用 |