|
@@ -4,7 +4,7 @@ import os
|
|
|
import time
|
|
import time
|
|
|
import sys
|
|
import sys
|
|
|
import traceback
|
|
import traceback
|
|
|
-os.environ["CUDA_VISIBLE_DEVICES"] = "-1"
|
|
|
|
|
|
|
+os.environ["CUDA_VISIBLE_DEVICES"] = "0"
|
|
|
import tensorflow as tf
|
|
import tensorflow as tf
|
|
|
tf.compat.v1.enable_eager_execution()
|
|
tf.compat.v1.enable_eager_execution()
|
|
|
MAX_COMPUTE = False
|
|
MAX_COMPUTE = False
|
|
@@ -26,7 +26,7 @@ if not MAX_COMPUTE:
|
|
|
|
|
|
|
|
os.environ['CUDA_CACHE_MAXSIZE'] = str(2147483648)
|
|
os.environ['CUDA_CACHE_MAXSIZE'] = str(2147483648)
|
|
|
os.environ['CUDA_CACHE_DISABLE'] = str(0)
|
|
os.environ['CUDA_CACHE_DISABLE'] = str(0)
|
|
|
- gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction=0.1)
|
|
|
|
|
|
|
+ gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction=0.05)
|
|
|
sess = tf.Session(config=tf.ConfigProto(gpu_options=gpu_options))
|
|
sess = tf.Session(config=tf.ConfigProto(gpu_options=gpu_options))
|
|
|
|
|
|
|
|
sys.path.append(os.path.dirname(os.path.abspath(__file__)) + "/../")
|
|
sys.path.append(os.path.dirname(os.path.abspath(__file__)) + "/../")
|
