ml4co_kit.utils.env_utils

The utilities used to install the environment.

Classes

EnvChecker()

EnvInstallHelper([pytorch_version, ...])

class ml4co_kit.utils.env_utils.EnvChecker[source]

Bases: object

check_cp310_or_later() bool[source]
check_cp311_or_later() bool[source]
check_cp39_or_later() bool[source]
check_cuda() bool[source]
check_dreamplace() bool[source]
check_gnn4co() bool[source]
check_gurobi() bool[source]
check_torch() bool[source]
class ml4co_kit.utils.env_utils.EnvInstallHelper(pytorch_version: str = '2.1.0', use_cuda: bool = False, cuda_version: str = '121')[source]

Bases: object

install()[source]