1.首先需要确认是否成功安装cuda，代码见图一；打印结果如图二所示。
图一
图二
2.如果未安装成功可以自行搜索，不麻烦；安装成功后需要分三步设置使用GPU，以简单的softmax分类器为例：
a.导入os模块
import os
#指定参与运算的显卡为GPU1，这个需要自己进系统管理器查看自己的显卡是GPU序号
os.environ['CUDA_LAUNCH_BLOCKING'] = '1'b.将模型放进GPU中运算。
c.更改训练、测试两个步骤，使用GPU运算。
文章出处登录后可见！
已经登录？立即刷新
