BUGFIX: now using correct paths to torch.