【TensorFlow】GPU和显存资源的配置
随时更新……使用指定的GPUGPU的编号在终端输入nvidia-smi查看。(1)在终端设置# 设置运行该程序可见的GPUCUDA_VISIBLE_DEVICES = 0,1,2 python your_file.py # 指定0#,1#,2#GPUCUDA_VISIBLE_DEVICES = "3,4" python your_file.py # 指定3#,4#GPUCUDA_VISIBLE_DEVICES = "" # 不使用GPU(2)在Python代码设置设置.
原创
2020-10-13 15:59:12 ·
383 阅读 ·
0 评论