python怎么利用gpu加速-木庄网络博客

2021-10-04 20:23:43 php中文网 Python 百度已收录

本文摘自php中文网，作者(*-*)浩，侵删。

看了好多教程都提到了使用 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即可

相关阅读 >>

更多相关阅读请进入《Python》频道 >>

书籍

Python编程从入门到实践第2版

￥69.8元 人民邮电出版社

python入门书籍，非常畅销，超高好评，python官方公认好书。