#question Hello. There was a problem deploying to streamlit. I did the same as in the video but it says something wrong.
I wrote the code correctly, I used venv
. but for some reason it is not working.
enter image description here
My GitHub profile shows the problem in full.
Run the code in kaggle or google colab notebooks. Then run this !nvidia-smi
find you have cuda, it shows cuda version. and use device = torch.device("cpu")
.
When you training the model use model_0 = modelname().to("cuda")