法1:

CUDA_VISIBLE_DEVICES=0,1,2 python test.py

法1中,0,1,2为显卡的编号。可以选择多块或单个显卡

法2:

    import os
    os.environ["CUDA_VISIBLE_DEVICES"] = "0,1,2"

 

步骤一:环境变量设置(好像不是必须的

改变系统变量仅使目标显卡,编辑 .bashrc文件,在里面添加系统变量

export CUDA_VISIBLE_DEVICES=0,1

linux的bashrc文件在/home目录下,但是是一个隐藏bai文件,在文件管理器里面du按Ctrl+H即可显示(显示为.bashrc,前zhi面小点儿表示隐藏文件)。也可以直接利用terminal直接打开

 

法1:

    在程序开头设置

比如写
os.environ["CUDA_VISIBLE_DEVICES"] = '1'就是用第二块显卡运行

import os
os.environ["CUDA_VISIBLE_DEVICES"] = '0,1,2,3'

法2:

    在运行程序时指定

运行程序时使用命令行,来设置该程序可见的gpu:

CUDA_VISIBLE_DEVICES=0,1,2,3 python xxx.py 

0,1,2为显卡的编号。可以选择多块或单个显卡


原文链接:https://blog.csdn.net/weixin_37707670/article/details/109898095

Logo

更多推荐