search

marmotlab / PRIMAL2

Training code PRIMAL2 - Public Repo
MIT License
150 stars 59 forks source link

hellow！i have a issues，Does this not require GPU? #2

Closed orangezishuai closed 3 years ago

orangezishuai commented 3 years ago

I noticed that tensorflow version is 1.11.0, and I want to ask if it doesn't require GPU training? Do I not need to install the GPU version of tensorflow?

thank you very much

GilesLuo commented 3 years ago

Hi, To run the code (either testing code or training code), you don't theoretically have to install tensorflow-gpu. Remember to use CUDA_VISIBLE_DEVICES=-1 to disable GPU when you run the code. But still, using GPU is encouraged since training/testing PRIMAL2 might need days even on powerful CPU and GPU (e.g. 2080ti).

Hope that help Zhiyao

orangezishuai commented 3 years ago

Hi, To run the code (either testing code or training code), you don't theoretically have to install tensorflow-gpu. Remember to use CUDA_VISIBLE_DEVICES=-1 to disable GPU when you run the code. But still, using GPU is encouraged since training/testing PRIMAL2 might need days even on powerful CPU and GPU (e.g. 2080ti).

Hope that help Zhiyao

thank you so much