当前第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用什么软件写爬虫
windows上使用Python增加或删除权限的方法
Python中strip什么意思
Python中import有什么用法
Python正则表达式怎么学习？
Python统计字符串中数字个数
Python的2d绘图库matplotlib的安装方法及功能介绍
Python 实用函数进阶（更新中）
k-means算法在Python中的实现
用Python绘制图形的实例详解
更多相关阅读请进入《Python》频道 >>
人民邮电出版社
python入门书籍，非常畅销，超高好评，python官方公认好书。
转载请注明出处：木庄网络博客 » python怎么利用gpu加速
