安装Totchvision 时自动安装cpu版本的torch

最新推荐文章于 2025-06-15 23:27:32 发布

jiashucai

最新推荐文章于 2025-06-15 23:27:32 发布

阅读量1k

点赞数 1

CC 4.0 BY-SA版权

文章标签：深度学习 pytorch python

本文链接：https://blog.youkuaiyun.com/jiashucai/article/details/127250394

本文解决了在安装PyTorch时遇到的问题：已正确安装CUDA版本的torch和torchvision，但在尝试将数据移至GPU时出现错误。问题在于安装torchvision时会默认安装CPU版本的torch并覆盖原有GPU版本。解决方案是先安装torchvision。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

今天在安装pytorch时发现了一个问题：

我明明安装了cuda版本对应（我这里是11.3的Gpu版本）torch和torchvision但是在运行存储数据到GPU的操作他会报错，找了许多文章发现在安装torchvision时他会默认的安装cpu版本的torch并且覆盖安装过的GPU版本的torch,所以我就先安装torchvision（虽说有点慢但是可以解决问题）