[PyTorch] no CUDA-capable device is detected - Printable Version +- Python Forum (https://python-forum.io) +-- Forum: Python Coding (https://python-forum.io/forum-7.html) +--- Forum: Data Science (https://python-forum.io/forum-44.html) +--- Thread: [PyTorch] no CUDA-capable device is detected (/thread-25964.html) |
[PyTorch] no CUDA-capable device is detected - constantin01 - Apr-17-2020 I use PyTorch for machine learning. I am doing all this in Google Colab. (best_val_loss, best_sentence_level_model) = train_eval_loop(sentence_level_model, train_dataset, test_dataset, F.cross_entropy, lr=5e-3, epoch_n=10, batch_size=64, device='cuda', early_stopping_patience=5, max_batches_per_epoch_train=500, max_batches_per_epoch_val=100) I use 'cuda' for training, but here is a problem: RuntimeError: cuda runtime error (100) : no CUDA-capable device is detected at /pytorch/aten/src/THC/THCGeneral.cpp:50did somebody meet this issue? how did you solve it? |