默认使用0号GPU。如果需要使用其他GPU,需要人为指定。
以1号GPU为例:
终端:CUDA_VISIBLE_DEVICES=1 python my_script.py
代码:
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "1"
默认使用0号GPU。如果需要使用其他GPU,需要人为指定。
以1号GPU为例:
终端:CUDA_VISIBLE_DEVICES=1 python my_script.py
代码:
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "1"