Skip to content

Instantly share code, notes, and snippets.

@imneonizer

imneonizer/cuda_visible_devices.py

Created April 14, 2021 12:00
Show Gist options
  • Save imneonizer/3e3cd3904e67aeeca28af79a65dd97bc to your computer and use it in GitHub Desktop.
Save imneonizer/3e3cd3904e67aeeca28af79a65dd97bc to your computer and use it in GitHub Desktop.
Download ZIP
Raw
cuda_visible_devices.py
# use multiple gpu devices by index
import os
#GPU_id = '0'
GPU_id = '0,1,2,3'
os.environ['CUDA_VISIBLE_DEVICES'] = GPU_id
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment