- 移除不兼容的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)
结束。