set_device
torch_
sys_
cuda
0.0.0
In torch_
sys_
cuda::
cuda
torch_sys_cuda
::
cuda
Function
set_
device
Copy item path
Source
pub fn set_device(device: CudaDevice) ->
Result
<
()
,
CudaError
>