tensorflow 自主选择使用的gpu
程序员文章站
2022-06-16 23:15:12
...
如果你的计算机不止一个gpu,在使用时你想自主选择使用其中一个的话,可以在程序开头(所有import之后)加上下列语句:
os.environ['CUDA_VISIBLE_DEVICES'] = '0' #使用GPU 0
如果想同时使用其中几个也很容易,用逗号隔开:
os.environ['CUDA_VISIBLE_DEVICES'] = '0,1,2' #同时使用GPU 0,1,2
上一篇: JDBC初步 JDBC连接SQLServer 2008之心路历程
下一篇: 函数问题
推荐阅读
-
iOS中使用UIDatePicker制作时间选择器的实例教程
-
iOS App中UIPickerView选择栏控件的使用实例解析
-
自动化收集SQLSERVER诊断信息的工具选择及使用介绍
-
在python下使用tensorflow判断是否存在文件夹的实例
-
使用HTML5技术开发一个属于自己的超酷颜色选择器
-
sai魔棒选择工具怎么选区细小的细节? sai魔棒工具的使用方法
-
Android selector状态选择器的使用详解
-
如何使用C#将Tensorflow训练的.pb文件用在生产环境详解
-
iOS中使用UIDatePicker制作时间选择器的实例教程
-
h5中类jQuery选择器querySelector的使用解析