PyTorch圖文安裝教程（Win10），含遇到的問題及解決辦法

WSquareJ發表於2022-02-19
原文網址 : https://www.cnblogs.com/wwj321/p/15332522.html
PyTorchWin10

PyTorch安裝教程（Win10)，含遇到的問題及解決辦法

1. 環境準備

首先，開啟PyTorch官網：https://pytorch.org/get-started/locally/
向下可以看到有如下幾個條件：

  • Python：PyTorch僅支援Python3.x，不支援Python2.x；

  • Package Manager：

    • Anaconda：Anaconda可以去官網下載，在這裡貼一個圖文教程：https://blog.csdn.net/weixin_43715458/article/details/100096496

    • pip：先到官網上下載pip檔案

      1632482888144

      然後將此檔案進行解壓，開啟cmd，將路徑調整到解壓後的pip資料夾：

      1632483073426

      執行如下命令：

      python setup.py install

      1632483205720

      最後為pip新增環境變數，找到python下的scrips，新增至系統變數Path下，記得後面加上分號。(一般在C:\Program Files\Python35\Scripts)

      檢查是否安裝成功，在cmd中輸入pip，返回如下結果：

      1632483791924

  • CUDA (這個很早就安裝了，有些忘了)

2. 安裝

Anaconda安裝完成後，點選win選單裡，開啟：

1632485934351

執行如下指令，建立一個新環境：

conda create -n pytorch python=3.7

pytorch為環境名稱，python=3.7為使用的python版本

然後會提示是否要安裝一些包，輸入y即可；

img

這樣即為新建環境成功，然後輸入如下指令，啟用環境：

conda activate pytorch

1632486284224

這個過程可能會遇到的問題：

CondaHTTPError: HTTP 000 CONNECTION FAILED for url <https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/win-64/current_repodata.json>

連線錯誤，我這裡的解決辦法是：在C盤C:\Users\Administrator開啟“.condarc“檔案，修改其內容如下：

ssl_verify: true
show_channel_urls: true

channels:
 - http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/win-64/
 - http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/win-64/

然後再次開啟PyTorch官網，可以在這裡選擇好指定的版本，開啟生成指令中的這個網站：

pip3 install torch==1.9.1+cu102 torchvision==0.10.1+cu102 torchaudio===0.9.1 -f https://download.pytorch.org/whl/torch_stable.html

1632485130413

開啟網站後，根據指令中的版本號選擇whl檔案，比如我是cu102，torch=1.9.1，以及我使用的是python3.7；

1632485319173

然後在pytorch環境、上述下載檔案的儲存路徑下輸入：

pip install torch-1.9.1+cu102-cp37-cp37m-win_amd64.whl

1632481875165

至此安裝成功；

torch-1.9.1+cu102-cp37-cp37m-win_amd64.whl下載連結：https://pan.baidu.com/s/16mt9FZrMa5UrUqvgqV6MEA
提取碼：pthn

3. 驗證

驗證PyTorch是否安裝成功：

  • 進入Python，輸入import torch，不報錯

  • 輸入如下指令，有返回值，表面可以正確使用：

    x = torch.rand(5, 3)
print(x)

    輸出類似下圖

  • 輸入測試gpu和cuda是否可用：

    torch.cuda.is_available()

    如果可用輸出True；

1632481898622

相關文章