Error occurred when converting Pytorch model to GPU model

Kotlin
IOS
VB.Net
SQL
Node.js
Swift
Ruby
Go
Oracle
Git
Laravel
Mongodb
Asp.Net
More...

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:

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?

False
Django, Why does the system automatically generate an empty session? every time the request is made?
Use python to convert dict1 to dict2, in the example. I tried for a long time but couldn't find a solution.
How to self-start the php-fpm service when building a LNMP environment with docker?
Can an array in react state be bound to a dom state using an element in the array?
Wechat browser version 6.6.5 cannot prevent the springback of pull-up and drop-down to the top under ios system.
In express+hbs, how do you define the routed data as a variable on the page?
How to use symbol reference of Iconfont in dva
How to elegantly add data to JSON objects?
Using node to build video streaming server
The proportional width of the echarts graph is inconsistent.
React uses axios to call the background data. After success, how to jump to the page based on the returned value?
I don't understand what a linux command means. Please explain it.
Vue introduces multiple components in vue-router by nesting named views, and does not refresh the same reference in other pages.
Running nodejs reports fewer errors ")"

Whether the socket_read in php will be disconnected if it hasn't received the message for a long time.
The introduction of a third-party js library that depends on jq in vue has been reporting to we need Jquery, but my jquery is globally registered, how to solve it?
Webpack4 packaging error
How to implement thread-safe INSERT IF NOT EXIST? gracefully by sqlalchemy
Webpack didn't repackage after saving the code?