摘要:
PyTorch中,.to(device) 是一个非常重要的方法,用于将张量、模型等对象移动到指定的设备(如CPU或GPU)。 import torch import torch.nn as nn # 检查可用设备 device = torch.device('cuda' if torch.cuda. 阅读全文
posted @ 2025-11-19 17:55
wangssd
阅读(58)
评论(0)
推荐(0)
摘要:
uv作为项目的包管理工具被广泛应用: #在当前目录中创建uv环境 uv init #查看当前可用python uv python list #查看当前依赖树 uv tree # 添加包 uv add torch # 更新包 如torch uv add torch --upgrade-package 阅读全文
posted @ 2025-11-19 17:34
wangssd
阅读(41)
评论(0)
推荐(0)

浙公网安备 33010602011771号