Hi, I have a Windows PC with latest Nvidia GPU and wanted to go through the lessons locally. I followed all the install instructions (Miniconda, mamba, “-c pytorch -c nvidia -c fastai” etc.) for Windows and I installed fastbook using this command:
mamba install -c fastai fastbook. When I execute the learner.fine_tune() method:
learn = cnn_learner(dls, resnet34, metrics=error_rate) learn.fine_tune(1)
I see NaN for training and validation loss. I couldn’t find any posts about people running into this issue on fastai v2 - am I missing something here?
PS. I see pytorch 1.9 warning about my GPU.
*C:\Users\shekh\miniconda3\envs\fastbook\lib\site-packages\torch\cuda_init_.py:106: UserWarning: *
GeForce RTX 3080 with CUDA capability sm_86 is not compatible with the current PyTorch installation.
The current PyTorch install supports CUDA capabilities sm_37 sm_50 sm_60 sm_61 sm_70 sm_75 compute_37.
If you want to use the GeForce RTX 3080 GPU with PyTorch, please check the instructions at Start Locally | PyTorch