- 博客(1)
- 资源 (2)
- 收藏
- 关注
原创 TensorFlow训练模型,指定GPU训练,设置显存,查看gpu占用
1 linux查看当前服务器GPU占用情况:nvidia-smi 周期性输出GPU使用情况: (如设置每3s显示一次GPU使用情况) watch -n 3 nvidia-smi 效果如下: 2 指定GPU训练,使用CUDA_VISIBLE_DEVICES来指定 如果要指定第2块GPU训练,可以在python代码中如下指定:import os os.environ['CUDA...
2018-02-28 16:42:19 8243 1
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人