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

Check if pytorch cuda is avalable (AssertionError: Torch not compiled with CUDA enabled) #346

Open
zdenop opened this issue Jul 25, 2019 · 1 comment

Comments

@zdenop
Copy link

zdenop commented Jul 25, 2019

Please use pytorch as shown in docs, otherwise users can get AssertionError: Torch not compiled with CUDA enabled.

E.g. in app-seperation-semseg/Background-Grayscale.py

if torch.cuda.is_available():
    device = 'cuda'
else:
    device = 'cpu'

segment(dlab, './images/greyscale/oldman.png', show_orig=False, dev=device)
@maddhatters-dev
Copy link

The problem I ran into was, after compiling the files they were scattered, I found a couple folders in my user dir and others in the programs and programs 32. Also I had to move the protoc and test files into the lib folder, for some reason they were not connecting till I did that. After moving the files I hit test it ran right through all of them.

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