tensorflow显卡内存不足||CUDA_VISIBLE_DEVICES=1 python **.py

 

元学习论文总结||小样本学习论文总结

2017-2019年计算机视觉顶会文章收录 AAAI2017-2019 CVPR2017-2019 ECCV2018 ICCV2017-2019 ICLR2017-2019 NIPS2017-2019

 

 

方法1 在python程序中

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

 方法2,运行时 

CUDA_VISIBLE_DEVICES=1 python **.py
 

你可能感兴趣的:(深度学习,TensorFLow)