e70a1e3fddeb24c02ea56f74ff4ce33c8642ac0c,autokeras/utils.py,,get_device,#,47

Before Change

            //       Free                 : xxxxxx MiB
            //                      ....
            smi_out = subprocess.check_output("nvidia-smi -q -d Memory | grep -A4 GPU|grep Free", shell=True)
            print(smi_out.decode("utf-8"))
        except subprocess.SubprocessError:
            warnings.warn("Cuda device successfully detected. However, nvidia-smi cannot be invoked")
            return "cpu"

After Change

            //       Free                 : xxxxxx MiB
            //                      ....
            smi_out = subprocess.check_output("nvidia-smi -q -d Memory | grep -A4 GPU|grep Free", shell=True)
            if isinstance(smi_out, bytes):
                smi_out = smi_out.decode("utf-8")
            print(smi_out)
        except subprocess.SubprocessError:
            warnings.warn("Cuda device successfully detected. However, nvidia-smi cannot be invoked")
            return "cpu"

Instances