如何指定程序在某个GPU上运行 python

分类: 文章 • 2025-03-12 22:18:09

只需要添加如下两行代码：

import os
os.environ["CUDA_VISIBLE_DEVICES"] = "1"

注意，服务器看到的GPU序号和手动在Linux上查看的序号是相反的，比如

那么我们指定os.environ[“CUDA_VISIBLE_DEVICES”] = “0”的话，你会发现在watch nvidia-smi指令下，2号GPU正在满负荷的运行！
---------------------

转载
原文：https://blog.****.net/qq_21460525/article/details/73477298