1.首先需要确认是否成功安装cuda,代码见图一;打印结果如图二所示。
图一
文章来源地址https://www.toymoban.com/news/detail-572790.html
图二
2.如果未安装成功可以自行搜索,不麻烦;安装成功后需要分三步设置使用GPU,以简单的softmax分类器为例:
a.导入os模块
import os #指定参与运算的显卡为GPU1,这个需要自己进系统管理器查看自己的显卡是GPU序号 os.environ['CUDA_LAUNCH_BLOCKING'] = '1'
b.将模型放进GPU中运算。
c.更改训练、测试两个步骤,使用GPU运算。
文章来源:https://www.toymoban.com/news/detail-572790.html
到了这里,关于python运行使用gpu运算【python基础】python开启GPU加速的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!