Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Run with GPU #56

Open
tejasa97 opened this issue Jun 9, 2019 · 4 comments
Open

Run with GPU #56

tejasa97 opened this issue Jun 9, 2019 · 4 comments

Comments

@tejasa97
Copy link

tejasa97 commented Jun 9, 2019

How does one run this code with GPU?

@ashnakapoor07
Copy link

Can I take this up as my first issue?

@tejasa97
Copy link
Author

@ashnakapoor07 Correct me if I'm mistaken, but I'm not sure anybody requires permission. Knock yourself out!

@titu1994
Copy link
Owner

If you have CUDA setup with Tensorflow / Theano, the scripts should automatically use the GPU.

If you do not have a local GPU, the colab link in the readme points to a GPU enabled notebook which you can use the script from, just run the cells one by one, select the content and style image and the options and there you go.

Also, @ashnakapoor07 Theres no need to request permission to resolve an issue. Go right ahead.

@f1am3d
Copy link

f1am3d commented Jul 29, 2019

@tejasa97

  1. Check tensorflow-gpu version that this project is using.
  2. Check CUDA compatibility with tensorflow-gpu version.
  3. Install needed CUDA version.
  4. Check system paths variables for CUDA installation.
  5. Install CUDA-compatible cuDNN version.
  6. Reboot your PC.
  7. Run from tensorflow.python.client import device_lib print(device_lib.list_local_devices())

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

4 participants