DirtyHarryLYL / HAKE-Action-Torch

HAKE-Action in PyTorch

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Basic environments of the branch

desenzhou opened this issue · comments

Hi, It's a wonderful work! I just want to know the basic environments of the branch, e.g. python version, cuda version etc. Would you please provide me the infomation? Thanks!

Please refer to the requirements in this conda configuration file: requirement.
You could simply use conda to install all the requirements following this list.
Python: 2.8.1 (

python-dateutil==2.8.1
).
As for cuda, simplify-x needs 10.0, and for our main model in TF 8.0 is enough.

So you could first run simplify-x with 10.0 and then change the version back to 8.0 to run the main project.

@DirtyHarryLYL Thanks for the reply. I will try it!