你即將離開本站

並前往https://github.com/NVIDIA/TensorRT/issues/1050

How to set cuda device with tensorRT python API? · Issue #1050

import pycuda.driver as cuda import pycuda.autoinit. e.g., to set os.environ['CUDA_DEVICE'] = '1' will make GPU:1 as the default device.

確定！ 回上一頁

查詢 「tensorrt pycuda」的人也找了：

  1. TensorRT
    2.
  2. tensorrt安裝
    3.
  3. Pycuda
    4.
  4. Pycuda install
    5.
  5. TensorRT install
    6.
  6. TensorRT TX2
    7.
  7. TensorRT Python
    8.
  8. TensorRT docker
    9.