指定GPU训练模型

指定GPU训练模型

1、直接在终端运行时加入相关语句实现指定GPU的使用

运行test.py文件时,使用编号为0的GPU卡

CUDA_VISIBLE_DEVICES=0 python test.py

运行test.py文件时,使用编号为0和2的GPU卡

CUDA_VISIBLE_DEVICES=0,2 python test.py

2、在Python程序中添加

使用第一张与第二张GPU卡

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

观察GPU使用情况

每隔0.1s刷新一下GPU运行状态

watch -n 0.1 nvidia-smi

你可能感兴趣的:(AI,深度学习)