错误一览
RuntimeError: No CUDA GPUs are available
这个问题比较常见了,原因在于源代码需要多块gpu,而我们的电脑通常只有1块,参考:在这里我也是将源代码中的’2’改为了’0’:
# os.environ["CUDA_VISIBLE_DEVICES"] = '2'
os.environ["CUDA_VISIBLE_DEVICES"] = '0'
TypeError:load() missing 1 required positional argument:‘Loader’
这个问题也比较常见,csdn上一搜一大堆,该问题在于使用yaml.load()时报错。原因是YAML 5.1版本后禁用了这个