seakafuka

> 我试过4060 mobile 显卡用官网的命令安装的执行训练会卡住，查看没有调用到 gpu。在 Tested with Torch 2.1 and CUDA 11.8, it works! #249 #249 的提示下 ，用命令`conda install pytorch==2.1.2 torchvision==0.16.2 torchaudio==2.1.2 pytorch-cuda=11.8 -c pytorch -c nvidia` 装了可以正常训练了。不知道是否官网的版本（pytorch==1.6.0 cudatoolkit=10.1）不支持40xx 的...

i correct it, because the book is reverse