報錯信息:
1.代碼頭部添加:
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "1"
2.限制當前腳本可用顯存
gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction=0.333)
sess = tf.Session(config=tf.ConfigProto(gpu_options=gpu_options))
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "1"
gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction=0.333)
sess = tf.Session(config=tf.ConfigProto(gpu_options=gpu_options))
增加你的ubuntu主機名到/etc/hosts中,此處以主機名aiserver爲例,具體如下: sudo vi /etc/hosts 如下圖所示添加127.0.0.1 aiserver即可