The current PyTorch install supports CUDA capabilities sm_37 sm_50 sm_60 sm_61 ... CPU: Intel i7-10700. half() # convert a model to 16-bit input = input.
確定! 回上一頁