Problem in running yolov8 on gpu with pytorch cuda()

Hi,

For torch 2.0.0+nv23.5, please use 0.14.1 TorchVision.
The compiling instructions can be found below:

Thanks.