在服务器虚拟环境中运行程序,tensorflow会默认占用所有的资源,可以在主程序中添加以下代码:
import tensorflow as tf
import os
os.environ["CUDA_VISIBLE_DEVICES"] = '1'
config = tf.ConfigProto()
config.gpu_options.per_process_gpu_memory_fraction = 0.5
config.gpu_options.allow_growth = True
sess = tf.Session(config = config)
再次运行又出现错误:SyntaxError: Non-ASCII character ‘\xe6’ in file amazon_wd.py on line 7, but no encoding declared; see http://python.org/dev/peps/pep-0263/ for details
解决办法:只需在文件开头添加以下命令:
# -*- coding: utf-8 -*