你即將離開本站

並前往https://pytorch-cn.readthedocs.io/zh/latest/notes/cuda/

CUDA语义 - PyTorch中文文档

CUDA语义. torch.cuda 会记录当前选择的GPU，并且分配的所有CUDA张量将在上面创建。可以使用 torch.cuda.device 上下文管理器更改所选设备。

確定！ 回上一頁

查詢 「pytorch指定gpu」的人也找了：

  1. 無法 指定GPU
    2.
  2. tensorflow指定gpu
    3.
  3. Pytorch使用GPU
    4.
  4. pytorch多gpu
    5.
  5. PyTorch CPU,GPU
    6.
  6. Pytorch to(device)
    7.
  7. Pytorch 限制GPU
    8.
  8. Pytorch model to(device)
    9.