python怎样应用gpu加快【Python教程】,python

看了很多教程都提到了运用 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练习，能够如许设置：

def set_gpus(gpu_index):
    if type(gpu_index) == list:
        gpu_index = ','.join(str(_) for _ in gpu_index)
    if type(gpu_index) ==int:
        gpu_index = str(gpu_index)
    os.environ["CUDA_VISIBLE_DEVICES"] = gpu_index

经由过程挪用set_gpu函数即可完成。

肯定要注意的是，CUDA_VISIBLE_DEVICES，每！一！个！字！母！ 必需完整写对，假如不小心写成了CUDA_VISIBLE_DIVICES或许CUDA_VISIABLE_DEVICES ，代码是不会报错的，然则GPU挪用不会胜利，

以上就是python怎样应用gpu加快的细致内容，更多请关注ki4网别的相干文章！