1.CPU training is not allowed.
本来是用pip install requirements_win.txt,自动装的torch,似乎是2.1.0,理论上是换成GPU版本的pytorch。先试试PyTorch
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
跑起来了
PS 显卡是影驰3060塑料大师MAxoc 12g,目前用了50条,大约一条5S的数据,大概是11秒左右一条。
怎么不动啊,3060,你快动啊,训练要训不动了。