欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

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
相关标签: gpu