Got stuck at model.to(cuda) , CUDA8.0

vision
1

I tried to train a neural network using Pytorch. However, the code was stuck at model.to(cuda). I am using CUDA 8.0 and try pytorch 1.0.0 and 0.4.1 with same problem. I checked the driver and it is good. And I tried the model.to(torch.device(cuda)). Could u pls help with me? Thank you very much.

2

conda list | grep pytorch
cuda80 1.0 h205658b_0 pytorch
pytorch 1.0.0 py3.7_cuda8.0.61_cudnn7.1.2_1 [cuda80] pytorch

3

nvidia-smi
Thu Jun 25 10:02:47 2020
±----------------------------------------------------------------------------+
| NVIDIA-SMI 375.26 Driver Version: 375.26 |
|-------------------------------±---------------------±---------------------+
| GPU Name Persistence-M| Bus-Id Disp.A | Volatile Uncorr. ECC |
| Fan Temp Perf Pwr:Usage/Cap| Memory-Usage | GPU-Util Compute M. |
|===============================+======================+======================|
| 0 GeForce GTX TIT… Off | 0000:83:00.0 Off | N/A |
| 27% 66C P0 93W / 250W | 1602MiB / 12206MiB | 100% Default |
±------------------------------±---------------------±---------------------+
| 1 GeForce GTX TIT… Off | 0000:84:00.0 Off | N/A |
| 22% 37C P8 17W / 250W | 1343MiB / 12206MiB | 0% Default |
±------------------------------±---------------------±---------------------+
| 2 GeForce GTX TIT… Off | 0000:87:00.0 Off | N/A |
| 38% 80C P2 173W / 250W | 9726MiB / 12206MiB | 79% Default |
±------------------------------±---------------------±---------------------+

±----------------------------------------------------------------------------+
| Processes: GPU Memory |
| GPU PID Type Process name Usage |
|=============================================================================|
| 2 22922 C python3 9724MiB |
±----------------------------------------------------------------------------+

4

Try updating CUDA to 9.2 or 10 with the latest stable pytorch