Steel
|
7e3c287f24
|
Create gpu-switch.sh
gpu switch version: 0.1
# Without GPU:
NVIDIA_VISIBLE_DEVICES="" ./gpu-switch.sh start container_name
# With single GPU:
NVIDIA_VISIBLE_DEVICES="0" ./gpu-switch.sh start container_name
# With multiple GPUs:
NVIDIA_VISIBLE_DEVICES="0,1" ./gpu-switch.sh start container_name
# With PCI addresses:
NVIDIA_VISIBLE_DEVICES="0000:03:00.0,0000:04:00.0" ./gpu-switch.sh start container_name
# Explicitly disable GPU:
NVIDIA_VISIBLE_DEVICES="none" ./gpu-switch.sh start container_name
|
2024-10-30 08:19:44 -05:00 |
|