查看机器上GPU情况
命令: nvidia-smi
功能:显示机器上gpu的情况
命令: nvidia-smi -l
功能:定时更新显示机器上gpu的情况
命令:watch -n 3 nvidia-smi
功能:设定刷新时间(秒)显示GPU使用情况

其中左上侧有0、1、2、3的编号,表示GPU的编号,在后面指定GPU时需要使用这个编号。
在终端执行程序时指定GPU
CUDA_VISIBLE_DEVICES=1 python your_file.py
这样在跑你的网络之前,告诉程序只能看到1号GPU,其他的GPU它不可见
可用的形式如下:
CUDA_VISIBLE_DEVICES=”” No GPU will be visible
以上内容我也没看明白!,,,,然后我就试试了下边的把代码粘在python程序的开头!然后就可以调用gpu了,不过我已经设置过gpu了,不知道为什么测rbf.nn这个神经网络代码时候,没调用我的gpu!
在Python代码中指定GPU
import os
设置定量的GPU使用量
设置最小的GPU使用量
发布者:全栈程序员-站长,转载请注明出处:https://javaforall.net/173913.html原文链接:https://javaforall.net
