Error reporting
indices should be either on cpu or on the same device as the indexed tensor (cpu)
reason
wrong version
solve
replace the correct version
conda install pytorch==1.12.1 torchvision==0.13.1 torchaudio==0.12.1 cudatoolkit=11.6 -c pytorch -c conda-forge
Post comment 取消回复