您的位置:首页 > 其它

深度学习实践笔记(不断更新)

2018-03-19 19:47 441 查看

写在前面:

本文主要是深度学习训练时候的一些经验和用法的笔记。
1、TensorFlow训练时指定GPU
    在终端执行程序时指定GPU   
     CUDA_VISIBLE_DEVICES=0,1   python  your_file.py    在Python代码中指定GPU     import os

     os.environ["CUDA_VISIBLE_DEVICES"] = "0"    设置定量的GPU使用量      config = tf.ConfigProto() 
      config.gpu_options.per_process_gpu_memory_fraction = 0.9 # 占用GPU90%的显存 

      session = tf.Session(config=config)
    设置最小的GPU使用量      config = tf.ConfigProto() 
      config.gpu_options.allow_growth = True 

      session = tf.Session(config=config)
2、最近使用jupyter notebook时出现no module named cv2,解决办法 pip install opencv-python
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: