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
>