安装pytorch(win 11 +3060 laptop)
- 移除不兼容的pytorch版本
运行代码时报错

这个错误表明正在使用的 PyTorch 版本过旧,它不支持 torch.utils.data.DataLoader 的 persistent_workers 参数或属性。
卸载旧版
pip uninstall torch torchvision torchaudio
按提示按Y和ENTER就可
- 安装适合RTX3060 laptop的版本
检查你的环境和兼容性
打开命令行(CMD),运行 nvidia-smi 命令。
我的CUDA版本是12.7

打开官网
https://pytorch.org/get-started/locally/
选择安装方式,环境,版本等参数,自定生成安装命令
这里没有12.7版本的,选择12.6版本
pip3 install torch torchvision --index-url https://download.pytorch.org/whl/cu126

3G左右,安装比较慢

- 验证是否成功
创建一个python脚本
代码入下
import torch
# 1. 检查 CUDA 是否可用 (必须为 True)
print("CUDA Is Available:", torch.cuda.is_available())
if torch.cuda.is_available():
# 2. 打印 PyTorch 内部编译的 CUDA 版本 (应该显示 12.6)
print("PyTorch CUDA Version:", torch.version.cuda)
# 3. 打印 GPU 名称
print("GPU Name:", torch.cuda.get_device_name(0))
# 4. 快速测试:将一个 Tensor 移动到 GPU 上
x = torch.tensor([1.0, 2.0], device='cuda')
print("Test Tensor on GPU:", x)

结束。

浙公网安备 33010602011771号