script for test_pytorch
ayclove opened this issue · comments
OS : Windows 10
GPU: RTX 3090 Ti
Driver: 536.40
Configure preset: windows-vs2022
I get module import error (torch) when loading test_pytorch script in Mogwai.
Is there any way for Mogwai to refer any virtual environment?
I would appreciate If you can show me any example regarding this.
Yes, it is supported but we haven't added documentation yet as all of this is still in very early stages. Basically what you want to do is set the FALCOR_USE_SYSTEM_PYTHON option in cmake and run cmake from a shell that has your virtualenv set up. This will make cmake pick up the correct Python binaries and compile for that environment.
Thanks for the tip. I confirmed that it is working fine as you suggested.