Closed DalInar closed 2 years ago
Strange, when I run the script, I get the correct results. I am using PhiFlow from 2.2-develop and Jax 0.3.15 but I havn't encountered this problem with any other version so far. Could you try with the latest version of Φ-Flow?
pip uninstall phiflow
pip install git+https://github.com/tum-pbs/PhiFlow@2.2-develop
Hello! I'm running the simple smoke plume demo from the examples and tried comparing the results with phi.torch vs phi.jax. The top gif shows the results I'm getting with phi.jax, and the bottom show the results with phi.torch. The only difference between the runs is
from phi.jaximport flow
vsfrom phi.torch import flow
. The results with torch seem correct, since they agree with the CPU results. My setup is Python 3.8.10, jax==0.3.14, jaxlib==0.3.14+cuda11.cudnn82, torch==1.12.0+cu116, cuda 11.6.Am I doing something wrong here?
Source code: