如果你的计算机不止一个gpu,在使用时你想自主选择使用其中一个的话,可以在程序开头(所有import之后)加上下列语句:
os.environ['CUDA_VISIBLE_DEVICES'] = '0' #使用GPU 0
如果想同时使用其中几个也很容易,用逗号隔开:
os.environ['CUDA_VISIBLE_DEVICES'] = '0,1,2' #同时使用GPU 0,1,2
如果你的计算机不止一个gpu,在使用时你想自主选择使用其中一个的话,可以在程序开头(所有import之后)加上下列语句:
os.environ['CUDA_VISIBLE_DEVICES'] = '0' #使用GPU 0
如果想同时使用其中几个也很容易,用逗号隔开:
os.environ['CUDA_VISIBLE_DEVICES'] = '0,1,2' #同时使用GPU 0,1,2