python如何使用gpu_【python基础】python开启GPU加速

前言

训练时使用GPU可以加速程序运行，本文介绍如何使用GPU加速。

前提条件

1. 机子有GPU显卡，并安装GPU显卡驱动；

2. 安装GPU的使用环境，CUDA等；

3. 打开nvidia-smi中的PM属性；

4. 程序中指定使用的GPU设备；

本文主要讲解如何指定GPU设备开启GPU进行加速。

操作过程

方法一：

看了好多教程都提到了使用 os.environ["CUDA_VISIBLE_DEVICES"] = "1" 或者export CUDA_VISIBLE_DEVICES=1，但是我加进代码里面速度并没有提高，查了很久才找到问题所在，当你的电脑有两块以上GPU时，上面这两句代码才起作用！如果电脑只有一块GPU，需要将参数的“1”改成“0”才可以，否则找不到“1”号设备，它就会默认使用CPU，速度不会提高。

方法二：

如果是在终端中运行python程序，使用命令：CUDA_VISIBLE_DEVICES=0 python filename.py即可

方法三：

如果有多块卡，想要指定多个gpu训练，可以这样设置：

defset_gpus(gpu_index):if type(gpu_index) ==list:

gpu_index= ','.join(str(_) for _ ingpu_index)if type(gpu_index) ==int:

gpu_index=str(gpu_index)

os.environ