ai_platform_nlu/nlp/__init__.py

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'