如何查看自己的电脑/服务器支持CUDA（torch.cuda.is_available）

Song • 1344 次浏览 • 0个评论 • 2018年07月11日

CUDA，是显卡厂商NVIDIA推出的运算平台，在Pytorch中我们会使用到CUDA，如何查看我们的服务器是否支持CUDA呢？其实很简单：[code]import torch print torch.cuda.is_available if torch.cuda.is_available(): print "support" else: print "not support"[/code]这样就可以知道自己的电脑是否支持CUDA了： [attach]30[/attach]

提交评论

请登录后评论

用户评论

    当前暂无评价,快来发表您的观点吧...

更多相关好文

    当前暂无更多相关好文推荐...

    公告提示

    Pytorch是Facebook的AI研究团队发布了一个Python工具包，是Python优先的深度学习框架。作为numpy的替代品；使用强大的GPU能力，提供最大的灵活性和速度,实现了机器学习框架Torch在Python语言环境的执行,基于python且具备强大GPU加速的张量和动态神经网络。