当前第2页 返回上一页
如果有多块卡，想要指定多个gpu训练，可以这样设置：
1 2 3 4 5 6 | 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加速的详细内容，更多文章请关注木庄网络博客！！
返回前面的内容
相关阅读 >>
使用Python可以做什么
安装Python后如何打开
Python中如何使用jpype实现调用java的实例分享
如何查看Python内置函数源码
怎么卸载Python 3.6？
Python如何使用xlrd实现读取合并单元格
Python中的析构函数详解
Python判断闰年程序是什么
Python用什么软件
Python中数据类型时间的介绍（附代码）
更多相关阅读请进入《Python》频道 >>
人民邮电出版社
python入门书籍，非常畅销，超高好评，python官方公认好书。
转载请注明出处：木庄网络博客 » python怎么利用gpu加速
