服务器有多台GPU 如何指定GPU运行程序
方法1:直接在运行程序命令 前添加 CUDA_VISIBLE_DEVICES=1(自己服务器上的GPU number) 样例如下：CUDA_VISIBLE_DEVICES=1 python your_program.py方法2：在自己的代码中添加import osos.environ["CUDA_VISIBLE_DEVICES"] = "2"...
·
方法1:直接在运行程序命令 前添加 CUDA_VISIBLE_DEVICES=1(自己服务器上的GPU number) 样例如下：
CUDA_VISIBLE_DEVICES=1 python your_program.py方法2：在自己的代码中添加
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "2"
技术共进，成长同行——讯飞AI开发者社区
更多推荐
2
0- 0
已为社区贡献1条内容
所有评论(0)