Instance type for cuda enabled pytorch

[beetlejuice007]

Hi

What are the different instance types I can use for a cuda enabled pytorch.

Thanking you

[aBurmeseDev]

Hi @hemantgujar - thanks for reaching out. I'm not quite sure if that's AWS CLI related but I could help point you to the right direction if you could elaborate your question for me to find the right resource?

Best,
John

[beetlejuice007]

I am trying to install cellbender (https://github.com/broadinstitute/CellBender). In one of the step I need to install pytorch (https://pytorch.org/get-started/locally/). So I can use the --cuda option.

After installation I should get

python3
import torch
print(torch.cuda.is_available())
TRUE

But this is not working for me.

I have checked for resources online:
https://stackoverflow.com/questions/60987997/why-torch-cuda-is-available-returns-false-even-after-installing-pytorch-with (I could not get enough information on type of nvidia graphic card on aws) and https://dziganto.github.io/aws/cuda/deep%20learning/gpu/python/pytorch/CUDA-on-AWS-for-Deep-Learning/ (seems like a very old publication didn't work for me).

Can you please help. Thanks.

[aBurmeseDev]

Thanks for sharing additional details.

I'm not sure how much of help I can provide since this isn't related to AWS CLI but here are the resources that I found within AWS regarding to what you're trying to accomplish:

• CUDA Installations and Framework Bindings
• Recommended GPU Instances

I would suggest reaching out to aws/deep-learning-containers repo.

Hope that helps!

[github-actions[bot]]

Hello! Reopening this discussion to make it searchable.