search

Joldnine / joldnine.github.io

My github.io blog repo. https://joldnine.github.io
2 stars 1 forks source link

Frequently Used Commands for Machine Learning #38

Open Joldnine opened 5 years ago

Joldnine commented 5 years ago

GPU

# Nvidia GPU information and status check
$ sudo nvidia-smi

# Nvidia GPU information and status check with trend and metrics (recommended)
$ sudo nvidia-smi --query-gpu=timestamp,pstate,temperature.gpu,utilization.gpu,utilization.memory,memory.total,memory.free,memory.used --format=csv -l 1

# Refresh GPU info every 5 seconds
$ watch -n 5 nvidia-smi

# Check version of CUDA
$ nvcc -V

Check if Torch is using GPU In python shell:

import torch
torch.cuda.get_device_name(0)

Check if tensorflow is using GPU In python shell: