指定用第二个GPU运行代码：
import os os.environ['CUDA_VISIBLE_DEVICES']='1'
或者在命令行终端输入的命令前加上CUDA_VISIBLE_DEVICES= *
CUDA_VISIBLE_DEVICES=0 python your_file.py
