python开启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训练，可以这样设置：

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
1
2
3
4
5
6
通过调用set_gpu函数即可实现。

Attention：
一定要注意的是，CUDA_VISIBLE_DEVICES，每！一！个！字！母！必须完全写对，如果不小心写成了 CUDA_VISIBLE_DIVICES 或者 CUDA_VISIABLE_DEVICES ，代码是不会报错的，但是GPU调用不会成功，排查问题浪费了我好几个小时。。。切记！
---------------------
作者：aaon22357
来源：CSDN
原文：https://blog.csdn.net/aaon22357/article/details/82733218
版权声明：本文为博主原创文章，转载请附上博文链接！

python开启GPU加速相关推荐

python运行使用gpu运算【python基础】python开启GPU加速
1.首先需要确认是否成功安装cuda,代码见图一:打印结果如图二所示. 图一图二 2.如果未安装成功可以自行搜索,不麻烦:安装成功后需要分三步设置使用GPU,以简单的softmax分类器为例: a. ...
air报错 Context3D not available. fd as3项目默认不开启gpu加速的问题
报错: Error #2044: Unhandled ErrorEvent:. text=Error #3702: Context3D not available. 默认没有开启gpu加速渲染, 到a ...
ae怎么设置gpu渲染_AE/PR mac版如何开启GPU加速渲染？
AE和PR是很多小伙伴比较熟悉的一款软件了,那么在使用 CC mac版的该软件的过程中如何开启GPU加速渲染呢?对此不清楚的小伙伴可以来看看小编为大家分享的方法啦~ 方法一: 在AE和PR的常规设置选 ...
matlab win10 gpu加速,win10怎么开启gpu加速有用吗 GPU硬件加速设置方法介绍
win10怎么开启gpu加速有用吗 GPU硬件加速设置方法介绍作为Windows 10的一项重要功能,微软在分享了GPU硬件加速细节后,现在还为大家送上了开启它的技巧,对于中低端设备来说,提升效果还 ...
gpu服务器接显示器缓慢,小科普开启GPU加速，又一波白给的性能提升？
每一波Win10更新,都带着一波新功能,那么代价是什么呢?就是新BUG,比方说今年2004版本刚出来的时候,老张的电脑只要装完重启就一定会卡"正在准备设备"后遇到DRIVER PN ...
如何在CST仿真软件中开启GPU加速
相信电磁领域的工作人员经常利用CST进行仿真运算,对于较大的模型和网格剖分,一般会在电脑里额外安装GPU加速卡.这里主要说明如下四个问题: 1.如何判断要购买哪种加速卡? 2.如何安装正确的驱动程序? ...
【Python - GPU】基于Python的GPU加速并行计算 -- pyCUDA
Python实现的CUDA – pyCUDA Nvidia的CUDA 架构为我们提供了一种便捷的方式来直接操纵GPU 并进行编程,但是基于 C语言的CUDA实现较为复杂,开发周期较长.而python ...
python opencv gpu加速_让Python下的OpenCV也能GPU加速!part.1
20200411更新: 经过评论区 @鹤汀凫渚的指导,我成功的用最简单的方法在python中调用到了GPU加速后的函数,这里把这位朋友的评论贴出来供各位参考: 以下原文: 本文的核心目的就是加速,在 ...
PS 开启GPU加速图片处理
还认为你的电脑的速度效果比不上苹果吗?还在嫌电脑渲染速度慢吗?试一下,电脑开启GPU硬件加速吧!只要有独显轻松加速(毕竟苹果笔记本要配独显电脑的价格基本上在15000以上,而且显卡的性能还不怎么好) ...

python开启GPU加速

python开启GPU加速相关推荐

最新文章

热门文章