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

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

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

你可能感兴趣的:(深度学习,pytorch,python)