Fix torch check

Change-Id: Iefd9737da4767b38d012df7834be972cccb1a8da
Signed-off-by: scott <scott@ubuntu>