gpu版本torch安装踩坑总结_torch gpu版本-CSDN博客

本文链接：https://blog.csdn.net/weixin_45456738/article/details/141029610

先写成功的办法：

0. 首先到torch官网上查看自己需要下载哪些对应版本的 .whl 文件

1. 在 pytorch官网上找到需要下载的`torch`、`torchvision`、`torchaudio`版本

在这里插入图片描述
我电脑 cuda 版本最高上限是12.0，所以选择下载 cuda11.8。我可以下载 torch2.2.0 的轮子文件。

2. 下载好 .whl 轮子文件

到 pytorch.whl 这里下载。比如我下载的 cuda 版本是11.8，python 版本是3.9，系统是 windows，我可以下载这些轮子（torchvision 和 torchaudio 下载很快，一秒就好了，torch下载很慢，要约50分钟）：
在这里插入图片描述

3. 下载了之后，进入到 cmd 中安装这些

记得去 cmd 中安装，不要去 anaconda prompt 里面安装。我也不知道为什么，我在 cmd 里面安装成功了，但是在 anaconda prompt 里面没成功。在 cmd 中进入到这几个 .whl 文件所在的文件夹内，以下红框是我存 .whl 的文件夹。
在这里插入图片描述
依次输入以下指令安装：

pip install torch-2.3.1+cu118-cp39-cp39-win_amd64.whl
pip install torchvision-0.18.1+cu118-cp39-cp39-win_amd64.whl
pip install torchaudio-2.3.1-cp39-cp39-win_amd64.whl

安装成功！图片如下。我不知道最后安装成功 torch 之后这几个 .whl 文件能不能删，我担心出错，就没有删。这个方法我之前就试过，但是之前没成功，报错在这里，不知道这次为啥成功了
在这里插入图片描述

踩过的坑：

我这次最开始是直接在 anaconda prompt 中下载 torch ，用的下面的 pip 命令，下载的实在是太慢太慢了，每次下载torch 都提示要将近三个小时才能下载下来，但是一般下了一小时就报错了。听 gpt 的换了好多次命令，也加了 timeout 时长，每次都这样失败，没卵用。建议多看几个别人的 torch 安装教程，不同的教程不同的方法总有一款适合你，别老是听 gpt 的钻到一种方法的死胡同里去了。
1.我最开始安装pytorch运行如下命令：

pip install torch==2.3.1+cu118 torchvision==0.18.1+cu118 torchaudio==2.3.1 --extra-index-url https://download.pytorch.org/whl/cu118 --timeout 1000

报错如下：

ERROR: THESE PACKAGES DO NOT MATCH THE HASHES FROM THE REQUIREMENTS FILE. If you have updated the package versions, please update the hashes. Otherwise, examine the package contents carefully; someone may have tampered with them.
    torch==2.3.1+cu118 from https://download.pytorch.org/whl/cu118/torch-2.3.1%2Bcu118-cp39-cp39-win_amd64.whl#sha256=78c9e0206f40a9f12c0369b2737d78d1998244238384286fd5492b39299059a7:
        Expected sha256 78c9e0206f40a9f12c0369b2737d78d1998244238384286fd5492b39299059a7
             Got        b6a7bc895b356aca75d5f385282dc75722e8be69a2ec96cd72932cdf8fa92900 //提示哈希值不匹配，通常是因为网速问题导致的对应的哈希值不匹配

2.torch官网上命令直接下载

conda install pytorch==2.2.0 torchvision==0.17.0 torchaudio==2.2.0 pytorch-cuda=11.8 -c pytorch -c nvidia

耗时太长，报错

CondaError: Downloaded bytes did not match Content-Length
  url: https://conda.anaconda.org/pytorch/win-64/pytorch-2.2.0-py3.9_cuda11.8_cudnn8_0.tar.bz2
  target_path: F:\Anaconda\pkgs\pytorch-2.2.0-py3.9_cuda11.8_cudnn8_0.tar.bz2
  Content-Length: 1518701418
  downloaded bytes: 29505897

3.conda换成pip

pip install pytorch==2.2.0 torchvision==0.17.0 torchaudio==2.2.0 pytorch-cuda=11.8 -c pytorch -c nvidia

又报错

ERROR: Could not open requirements file: [Errno 2] No such file or directory: 'pytorch'

4.conda版本太低了4.10.3，更新到了22.9.0，继续下载。
还是报错。

5.最终的解决办法就是轮子文件