lazyprogrammer / machine_learning_examples

A collection of machine learning examples and tutorials.
https://lazyprogrammer.me
8.37k stars 6.34k forks source link

Low GPU utilization #46

Open DongChen06 opened 5 years ago

DongChen06 commented 5 years ago

Hi, I use the code DQN, I can run the code successfully, but I find the GPU utilization is really low, about 8%. I use the 2080Ti and I use the GPU by adding the code at the begining of the DQN file.

import os

os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"
os.environ["CUDA_VISIBLE_DEVICES"] = "0"

How can I improve the GPU utilization? Thanks!!