4 lines
218 B
Python
4 lines
218 B
Python
import torch
|
|
print("CUDA 版本：", torch.version.cuda) # 应输出 12.8
|
|
print("cuDNN 版本：", torch.backends.cudnn.version()) # 应输出 9200+
|
|
print("GPU 可用：", torch.cuda.is_available()) # 应输出 True |