4 lines
127 B
Python
4 lines
127 B
Python
|
import os
|
||
|
|
os.environ['TF_KERAS'] = '1'
|
||
|
|
os.environ["CUDA_DEVICE_ORDER"] = 'PCI_BUS_ID'
|
||
|
|
os.environ['CUDA_VISIBLE_DEVICES'] = '-1'
|