import torch # 如果pytorch安装成功即可导入
print(torch.__version__) #torch版本
print(torch.cuda.is_available()) # 查看CUDA是否可用
print(torch.cuda.device_count()) # 查看可用的CUDA数量
print(torch.version.cuda) # 查看CUDA的版本号
深度学习环境搭建测试
最新推荐文章于 2024-05-18 22:06:58 发布