import torch
import torch.nn as nn
print(torch.cuda.is_available())# 如果输出是True,那意味着你的PyTorch安装已经正确配置了CUDA支持,如果是False,则表示CUDA不可用
print(torch.__version__) # 打印出当前安装的PyTorch版本
print(torch.cuda_version
检查GPU版本的Pytorch安装是否成功
最新推荐文章于 2025-12-26 12:09:14 发布

最低0.47元/天 解锁文章
4974

被折叠的 条评论
为什么被折叠?



