摘要: 输出关于PyTorch、CUDA设备以及CUDA运行时的相关信息 import torch def check_torch_and_cuda_details(): # 检查PyTorch版本 print("PyTorch version:", torch.__version__) # 检查CUDA是 阅读全文
posted @ 2024-03-25 15:48 阿坦 阅读(38) 评论(0) 推荐(0)