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

Torch CUDA #11

Closed
turbo opened this issue Dec 4, 2017 · 3 comments
Closed

Torch CUDA #11

turbo opened this issue Dec 4, 2017 · 3 comments

Comments

@turbo
Copy link

turbo commented Dec 4, 2017

IMO the install section should mention that pytorch needs to be compiled with CUDA support, which is quite fiddly on some macOS versions (needing to downgrade clang to compile at all, since nvcc doesn't like clang 90000).

@DmitryUlyanov
Copy link
Owner

You can change dtype to torch.FloatTensor at the top of each notebook to use CPU instead of GPU.

@turbo
Copy link
Author

turbo commented Dec 4, 2017

Oh I did manage to install it, it just might trip up some people who pip install torch without reading pytorch's instructions :)

@turbo turbo closed this as completed Dec 4, 2017
@turbo
Copy link
Author

turbo commented Dec 4, 2017

@DmitryUlyanov BTW, in inpainting, dtype is defined two times. That's probably a typo.

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

2 participants