ubuntu系统环境指定GPU的使用
1改变系统变量改变系统环境变量仅使目标显卡,编辑 .bashrc文件，添加系统变量export CUDA_VISIBLE_DEVICES=0linux的bashrc文件在/home目录下，但是是一个隐藏bai文件，在文件管理器里面du按Ctrl+H即可显示(显示为.bashrc，前zhi面小点儿表示隐藏文件）。也可以直接利用terminal直接打开1sudogedit～/.bashrcgedit换
·
法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，1linux的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
更多推荐
2
0- 0
已为社区贡献1条内容
所有评论(0)