原文：https://blog.csdn.net/strive_for_future/article/details/81292865?utm_source=copy

1）需要指定GPU，程式碼頭部新增如下程式碼：

import os
os.environ["CUDA_VISIBLE_DEVICES"] = "1"

2）限制當前指令碼可用視訊記憶體，程式碼頭部新增第一行，session 語句進行如第二行的修改

gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction=0.333)
sess = tf.Session(config=tf.ConfigProto(gpu_options=gpu_options))

 
 

你好博主，這個方法確實有效果

1）我方在了程式碼頭部

2）裡面的第一句話我用上了替換了原來的對應程式碼

請問最後的這句我怎麼在我的程式裡面沒有看到

sess = tf.Session(config=tf.ConfigProto(gpu_options=gpu_options))

下面是我的情況

全