本文共 118 字,大约阅读时间需要 1 分钟。
在用别人的代码跑实验时候,需要指定GPU来跑,但是代码里面又没有给GPU编号相关参数。
简单粗暴,直接在开头指定
CUDA_VISIBLE_DEVICES=0 python train.py
同时这也是官方建议的方法
转载地址:http://igrpi.baihongyu.com/