MMPretrain CUDA available: False

[L-A-Sandhu]

I am writing a code for image classification using a swin transformer. However, it says CUDA available: False. whereas, i have 3060 installed in the system that works well with TensorFlow and Pytorch frameworks.
Please check if my GPU is in use as my training crashes after first epoch

[Ezra-Yu]

try:

>>> import torch
>>> torch.cuda.is_available()
True

If the output is False, there maybe some thing wrong with your installation with your torch. please refer to the PyTorch for more help.