Thresh Frazier

icon location My name is Thresh Frazier, and I am a college student. I am interested in programming and mainly use Python. I am learning Vue to build my own web page

Results 2 comments of Thresh Frazier

请问怎么用GPU训练

train.py中修改： device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu") cnn = CNN().to(device) images = Variable(images).to(device) labels = Variable(labels.float()).to(device) ![屏幕截图 2021-02-23 103023](https://user-images.githubusercontent.com/21073721/108795942-7b706200-75c2-11eb-8f88-68525acb7768.png

直接就用cpu训练，把我的电脑cpu给爆了 >_<

尝试下在train.py中修改： device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")