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

redis.exceptions.ResponseError: GPU requested but Torch couldn't find CUDA #16

Open
Hanson0910 opened this issue Sep 11, 2020 · 3 comments

Comments

@Hanson0910
Copy link

Hi,When I use con.scriptset('imagenet_script', 'GPU', script) then raise redis.exceptions.ResponseError: GPU requested but Torch couldn't find CUDA.I had build the redisai with gpu.I can not fina any case with gpu in pyhton.

@hhsecond
Copy link

It's happening because an internal GPU check using torch is failing (a check similar to torch.cuda.is_available() from python). Could you provide the steps you have used to build RedisAI?

@Hanson0910
Copy link
Author

Thanks for your prompt reply,this is my step:
1)git clone --recursive https://github.com/RedisAI/RedisAI;
2)bash get_deps.sh, it will download tf and torch..
3)ALL=1 make -C opt clean build
I build them in the python Virtual environment with python3.6 and torch1.5.1,the redis version=5.0.0

@rafie
Copy link

rafie commented Sep 14, 2020

There's an error in the docs.
One should run the following to build RedisAI for GPU (that is, instead of steps 2, 3):

make -C opt fetch build GPU=1

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

3 participants