Error occurred when converting Pytorch model to GPU model

because there is no PyTorch tag at present, so TensorFlow is used. Would you like to add this tag to those with high reputation? Thanks tags ()

recently, I"m going to use the VGG model to train the data, and then I want to use GPU, but I always report an error when converting the VGG model to VGG.cuda (). At first, I thought it was a problem with the dataset, but then I converted VGG to .cuda () without doing anything, but I still reported an error. The code and error report are as follows:

Tensorflow Deep-Learning artificial-Intelligence Machine-Learning python
Feb.28,2021

hint: is there not enough video memory in out of memory,? I tried it, pytorch 0.3.1 + torchvision 0.2.0 to load VGG16 requires more than 700 MB memory:

clipboard.png

you can try a smaller model or see if there is a problem with cuda.

Previous: Redis backup recovery

Next: How does vscode automatically complete the introduction module?

css mysql arrays josn react html typescript webpack npm sass R objective-c .net sql-server jquery python-3.x angularjs django angular excel regex iphone ajax linux xml pandas vba spring database wordpress string wpf xcode windows bash postgresql oracle multithreading eclipse list firebase algorithm macos forms image scala visual-studio azure bootstrap spring-boot react-native python-2.7 docker performance function winforms matlab powershell apache dataframe api sqlite numpy rest shell selenium flutter dart maven loops qt swing android-studio csv express file class tensorflow sorting codeigniter perl
MySQL Query : SELECT * FROM `codeshelper`.`v9_news` WHERE status=99 AND catid='6' ORDER BY rand() LIMIT 5
MySQL Error : Disk full (/tmp/#sql-temptable-64f5-1b2c381-2bb02.MAI); waiting for someone to free some space... (errno: 28 "No space left on device")
MySQL Errno : 1021
Message : Disk full (/tmp/#sql-temptable-64f5-1b2c381-2bb02.MAI); waiting for someone to free some space... (errno: 28 "No space left on device")
Need Help?