ubuntu系统环境指定GPU的使用

法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