If you are running ubuntu, you can type nvidia-smi in your terminal after you run your code to see details of your gpu usage.
Also, to force fastai to use your gpu, you can do .cuda() at the end of your code.
If you are running ubuntu, you can type nvidia-smi in your terminal after you run your code to see details of your gpu usage.
Also, to force fastai to use your gpu, you can do .cuda() at the end of your code.