如何将ndarray转换为GPU格式的autograd变量?

时间:2019-03-28 12:30:40

标签: python numpy gpu pytorch autograd

我正在尝试做这样的事情,

try { $dbh = new PDO('mysql:host=localhost;dbname=banim', 'root', ''); $uName = "banim"; //$_POST['uName']; $email = "Rabak@gmail.com"; //$_POST['email']; $query = $dbh->prepare("SELECT * from users WHERE email = :email OR uName = :name"); $query->setFetchMode(PDO::FETCH_ASSOC); $query->bindParam(":name", $uName); $query->bindParam(":email", $email); $query->execute(); while($row = $query->fetch()){ print_r($row); } } catch (PDOException $e) { echo "PDOException: " . $e->getMssage() . PHP_EOL; }

它属于data = torch.autograd.Variable(torch.from_numpy(nd_array))类型,但是我也需要Variable[torch.FloatTensor],我也想在Variable[torch.cuda.FloatTensor]中做到这一点,它缺乏像pytorch version 0.3.0或{{1} }

1 个答案:

答案 0 :(得分:1)

您可以使用张量的cuda()方法。

如果您想使用特定的设备,则可以使用上下文管理器,例如

with torch.cuda.device(device_index):
    t = torch.FloatTensor(1.).cuda()

有关更多具体信息,check documentation for version 0.3.0