指定显卡运行python脚本

指定显卡运行python脚本

  • 指定显卡运行python脚本

指定显卡运行python脚本

 CUDA_VISIBLE_DEVICES=0 python infer.py

你可能感兴趣的:(pytorch)