查看cuda是否可用

三千院本院

于 2024-08-14 11:05:33 发布

阅读量390

点赞数 2

CC 4.0 BY-SA版权

文章标签：深度学习人工智能

本文链接：https://blog.csdn.net/weixin_42225889/article/details/141185523

import torch
print(torch.cuda.is_available())		# cuda是否可用
print(torch.version.cuda)				# 查看cuda版本
print(torch.cuda.device_count())		# 查看GPU数量
print(torch.__version__)				# 查看torch版本
print(torch.backends.cudnn.version())   # 检查cudnn版本