Fastai doesn't detect the gpu in kaggle issue

[bibs2091 (Mohamed Naas)]

Hello everybody,
I am trying to make a classification model in kaggle, kaggle is using the previous version of fastai, so i upgraded it along with torch,torchvision :

I import fastai :

when I check if am using the gpu :

it shows false that means it doesn’t use it
keep in mind that the gpu is activated

when i run this

i get this error :

Can you help me please !!
thank you in advance

[muellerzr (Zachary Mueller)]

You need to install the correct torch wheel for your CUDA version.

[bibs2091 (Mohamed Naas)]

can you provide a code sample ?

[muellerzr (Zachary Mueller)]

Quick search would have shown this:

Fastai v2 chat

Thanks. This is why I couldn’t get cuda support on Kaggle with PyTorch 1.6. Kaggle is running cuda 10.1, while PyTorch assumes you’re using 10.2 when using the default pip install. This works for me on Kaggle: !pip install torch==1.6.0+cu101 torchvision==0.7.0+cu101 -f https://download.pytorch.org/whl/torch_stable.html !pip install fastai2 Then when checking: import torch torch.cuda.is_available() Then output is True, meaning that GPU acceleration is working

[bibs2091 (Mohamed Naas)]

thanks it worked