自定义博客皮肤VIP专享

*博客头图：

点击选择上传的图片

格式为PNG、JPG，宽度*高度大于1920*100像素，不超过2MB，主视觉建议放在右侧，请参照线上博客头图

请上传大于1920*100像素的图片！

博客底图：

点击选择上传的图片

图片格式为PNG、JPG，不超过1MB，可上下左右平铺至整个背景

栏目图：

点击选择上传的图片

图片格式为PNG、JPG，图片宽度*高度为300*38像素，不超过0.5MB

主标题颜色：

RGB颜色，例如：#AFAFAF

Hover：

RGB颜色，例如：#AFAFAF

副标题颜色：

RGB颜色，例如：#AFAFAF

预览取消提交

自定义博客皮肤

-+

上一步保存

qq_18833967的博客

pytorch

关注

关注数：文章数：1 文章阅读量：3885 文章收藏量：7

作者: 在路上66

这个作者很懒，什么都没留下…

展开

判断pytorch是否可用？输出GPU数量和编号

1.输出torch版本号 2.判断安装后的torch是否可用？ 3.输出GPU数 4.输出GPU名字 5.输出现在正在使用的GPU编号 import torch print("torch版本号:",end="") print(torch.__version__) print("判断torch是否可用:",end="") print(torch.cuda.is_available()) print("gpu数量:",end="") print(torch.cuda.device_cou.

原创 2021-05-19 17:06:36 · 3885 阅读 · 0 评论