hi, I have solved this problem, in setup.py, the code assigns your pytorch version according to your machine statement. so you need to comment the version part.In the final step, you should install torch vision alone.