colab测试一下有没有使用GPU

表白：黑白圣堂血天使，天剑鬼刀阿修罗。 
讲解对象：/colab测试一下有没有使用GPU 
作者：融水公子 rsgz
Colab教程

Colab教程 http://www.rsgz.top/post/410.html

很多人虽然在 代码执行程序 那里设置了 GPU运行时  但是还是不敢确定有没有使用GPU

有没有什么办法可以测试呢？当然有，先查看模块列表里面的模块有没有torch

!pip list | grep torch

如果你没有打开GPU应该会是这样的
输出GPU的名字

import torchprint(torch.cuda.get_device_name(0))#输出GPU的名字===RuntimeError: No CUDA GPUs are available #没有使用GPU

查看一下是否有可用的GPU

print(torch.cuda.is_available())# False  无可用的GPU

这个时候打开GPU 然后再次尝试  成功了！

===

公众号：小雪妃

谢谢大家的支持！可以点击我的头像，进入我的空间浏览更多文章呢。建议大家360doc[www.360doc.com]注册一个账号登录，里面真的有很多优秀的文章，欢迎大家的到来。
---