服務器GPU查看和設置

查看GPU佔用情況 nvidia-smi 設置程序運行GPU 代碼中加入 import os os.environ["CUDA_VISIBLE_DEVICES"] = "1" 注意： 服務器看到的GPU序號和手動在Linux上查看的序號是相反的，比如 如果指定os.environ[「CUDA_VISIBLE_DEVICES」] = 「0」，則在watch nvidia-smi指令下，2號GPU正在

>>阅读原文<<