Pytorch: 指定运行 GPU
2019-09-19 本文已影响0人
Gavin先生
在终端使用如下命令查看，计算机本身的 GPU 数目及使用情况。
nvidia-smi
nvidia-smi_info.png
打印出的信息代表的意思：
GPU: GPU序列号，有几个GPU就显示几条。
Fan:风扇转速，从0到100%之间变动，这个速度是计算机期望的风扇转速，实际情况下如果风扇堵转，可能打不到显示的转速
Temp: 是温度，单位摄氏度。
Perf: 性能状态，从P0到P12，P0表示最大性能，P12表示状态最小性能。
Pwr:Usage/Cap: 是能耗。
Persistence-M: 是持续模式的状态，持续模式虽然耗能大，但是在新的GPU应用启动时，花费的时间更少。
Bus-Id: 涉及GPU总线的东西
Disp.A: Display Active，表示GPU的显示是否初始化。
Memory-Usage: 显存使用率。
GPU-Util: GPU利用率
然后根据GPU数目可以指定程序运行使用哪块 GPU，默认程序使用的是 GPU 0， 我们可以在代码最前面添加如下指令指定使用的GPU。
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "1"
