python运行使用gpu运算【python基础】python开启GPU加速

1.首先需要确认是否成功安装cuda，代码见图一；打印结果如图二所示。

 图一

 

图二 

2.如果未安装成功可以自行搜索，不麻烦；安装成功后需要分三步设置使用GPU，以简单的softmax分类器为例：

a.导入os模块

import os

#指定参与运算的显卡为GPU1，这个需要自己进系统管理器查看自己的显卡是GPU序号
os.environ['CUDA_LAUNCH_BLOCKING'] = '1'

b.将模型放进GPU中运算。

 c.更改训练、测试两个步骤，使用GPU运算。