From 864ae429ca461e960931be701c60287fd1d2228a Mon Sep 17 00:00:00 2001 From: Thiago Franco de Moraes Date: Tue, 16 Mar 2021 19:03:57 -0300 Subject: [PATCH] Plaidml was not working when installed in virtualenv on WIndows --- invesalius/segmentation/brain/utils.py | 8 ++++++++ 1 file changed, 8 insertions(+), 0 deletions(-) diff --git a/invesalius/segmentation/brain/utils.py b/invesalius/segmentation/brain/utils.py index b1d1bb6..3bc12b6 100644 --- a/invesalius/segmentation/brain/utils.py +++ b/invesalius/segmentation/brain/utils.py @@ -15,6 +15,14 @@ def prepare_plaidml(): if local_user_plaidml.exists(): os.environ["RUNFILES_DIR"] = str(local_user_plaidml) os.environ["PLAIDML_NATIVE_PATH"] = str(pathlib.Path("/usr/local/lib/libplaidml.dylib").expanduser().absolute()) + elif sys.platform == "win32": + if 'VIRTUAL_ENV' in os.environ: + local_user_plaidml = pathlib.Path(os.environ["VIRTUAL_ENV"]).joinpath("share/plaidml") + plaidml_dll = pathlib.Path(os.environ["VIRTUAL_ENV"]).joinpath("library/bin/plaidml.dll") + if local_user_plaidml.exists(): + os.environ["RUNFILES_DIR"] = str(local_user_plaidml) + if plaidml_dll.exists(): + os.environ["PLAIDML_NATIVE_PATH"] = str(plaidml_dll) def prepare_ambient(backend, device_id, use_gpu): if backend.lower() == 'plaidml': -- libgit2 0.21.2