---

前言

本篇文章主要介紹在Windows下 python 3.7 配置 pytorch，幫助需要的朋友避坑

安裝 pytorch 需要多個版本適配，本文提供一種使用于python 3.7 和 cuda 的安裝方法，同時給出一些處理問題的建議

---

一、安裝 python

python 3.7 是比較穩定的版本，可以根據自己的需求安裝，可以參考博客：anaconda安裝

• 補充：anaconda 歷史版本倉庫

二、安裝 cuda + cudnn

參考安裝博客：cuda安裝

補充：查看 cuda 版本 nvidia-smi

只要下載的 cuda 版本不高于自己顯卡支持的版本就行

二、安裝 pytorch

2.1 版本匹配

根據自己安裝的 cuda 去找對應的 pytorch、torchvision 和 torchaudio

2.1.1 方法一

去 pytorch 官網找對應的版本：官網地址

需要注意的是：torch、torchvision 以及 torchaudio 版本一定要對應，具體看官網給出的版本。比如，cuda 11.8 可以安裝 torch 2.0.1 + torchvision 0.15.2 + torchaudio 2.0.2

• 在換源后復制安裝命令進行安裝，如這里的 cuda 11.8
  
  如果順利的話，這一行命令就可以解決問題，但是可能因為版本不兼容等問題，這里出現的版本可能在倉庫中不存在。比如我要安裝的 python 3.7 + cuda 11.1 所有給出滿足的版本如下：
  
  cuda 11.1 + pytorch 倉庫地址
  倉庫地址大家可以根據自己要的 cuda 版本找，如下：
  
  一般而言在 cmd 中無法下載并報找不到對應版本錯誤，那么倉庫中就沒有

2.1.2 方法二

由于 python 3.7 + cuda 11.1 在上面所有版本中都出現了倉庫中找不到某一個庫的適配版本，比如能找到 torch 1.10.1，但是找不到 torchvision 0.11.2，意味著無法安裝。

解決辦法

• 卸載 python 3.7，安裝 python 3.8
• 卸載 cuda 11.1，裝一個低版本
• 都不用卸載，找清華鏡像或者國內其他鏡像，看是否有適合的版本 torch 的清華鏡像
  

補充：

• cu102 是指 cuda 10.2
• py37 是指 python 3.7
• cudnn 8_0 是指 cudnn 8.*

2.2 安裝 .tar.bz2

可以將下載的 .tar.bz2 文件放在 anaconda 的 pkgs 下，并在該目錄下打開命令行，安裝命令：conda install --use-local 包名

---

三、驗證是否安裝成功

cmd 進入 python 環境

import torch
print(torch.cuda.is_available())

總結

運氣不好裝 torch真的很需要耐心~~~~