pick_torch_device (utils/cuda_utils.py)

rocketride-server · packages/ai/src/ai/common/utils/cuda_utils.py · python · MIT
ai, tag:knn-classified

Utility functions to pick torch device, dtype, and estimate GPU memory.

# Source: https://github.com/rocketride-org/rocketride-server | License: MIT | Copyright 2026 Aparavi Software AG
# Provided via justclaw.it.com; keep this notice with the code.
# =============================================================================
# MIT License
#
# Copyright (c) 2026 Aparavi Software AG
#
# Permission is hereby granted, free of charge, to any person obtaining a copy
# of this software and associated documentation files (the "Software"), to deal
# in the Software without restriction, including without limitation the rights
# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
# copies of the Software, and to permit persons to whom the Software is
# furnished to do so, subject to the following conditions:
#
# The above copyright notice and thi

... (truncated -- full source via MCP)

See the full source, get the GitHub permalink, and search for more like it.

Get a free API key