首頁 > 軟體

解決anaconda安裝pytorch報錯找不到包的問題

2023-03-13 06:00:10

前言

使用anaconda 命令安裝pytorch包的問題,以下給出解決辦法。

狀況描述:

由於是第一次接觸安裝,看了很多篇文章,也遇到了一些坑,在此總結一下。

安裝pytorch,從硬體上分為2個版本,一個是cpu,一個是gpu(顯示卡)。gpu版本可以加速訓練效果。

由於本人的電腦比較舊,顯示卡是rtx1050,我安裝了cuda9.0版本+cudnn+pytorch(torch+torchvision)

本人安裝的是gpu版本的,在這個過程中,走了一些彎路。
大致流程是這樣:

  • 檢視顯示卡驅動,看支援什麼 版本的cuda
  • 下載安裝cuda,將cudnn解壓到cuda安裝目錄下
  • 安裝 anaconda,建立對應的python版本環境
  • 將pytorch 安裝進anaconda對應的環境中

總結:上述的具體流程可以參考其他博文,這裡就大致說一下。

問題描述

本人按照上述步驟走到最後一步發現報錯,
報錯如下:

conda install pytorch==1.1.0 torchvision==0.3.0 cudatoolkit=9.0 -c pytorch
Solving environment: failed

PackagesNotFoundError: The following packages are not available from current channels:

  - pytorch==1.1.0
  - torchvision==0.3.0

Current channels:

  - https://conda.anaconda.org/pytorch/win-64
  - https://conda.anaconda.org/pytorch/noarch
  - https://repo.anaconda.com/pkgs/main/win-64
  - https://repo.anaconda.com/pkgs/main/noarch
  - https://repo.anaconda.com/pkgs/free/win-64
  - https://repo.anaconda.com/pkgs/free/noarch
  - https://repo.anaconda.com/pkgs/r/win-64
  - https://repo.anaconda.com/pkgs/r/noarch
  - https://repo.anaconda.com/pkgs/pro/win-64
  - https://repo.anaconda.com/pkgs/pro/noarch
  - https://repo.anaconda.com/pkgs/msys2/win-64
  - https://repo.anaconda.com/pkgs/msys2/noarch

To search for alternate channels that may provide the conda package you're
looking for, navigate to

    https://anaconda.org

and use the search bar at the top of the page.

存取官網,發現官網上面的歷史版本最低都是 1.4在往下就沒有了。

解決辦法:

下載安裝包,本地 pip install。

pytorch下載連結
版本號對應查詢
假設:電腦已經下載好了 安裝包 (torch+torchvision)

開啟 Anaconda Prompt

切換到虛擬環境:activate pytorch (此處的pytorch是anaconda 新建立的)

切換碟符到 安裝包目錄下,使用命令pip install xxx.whl即可

舉例如下:

(pytorch) D:Anaconda3envspytorchLibsite-packageswheel>pip install torch-1.1.0-cp37-cp37m-win_amd64.whl
Looking in indexes: https://pypi.tuna.tsinghua.edu.cn/simple
Processing d:anaconda3envspytorchlibsite-packageswheeltorch-1.1.0-cp37-cp37m-win_amd64.whl
Collecting numpy
  Using cached https://pypi.tuna.tsinghua.edu.cn/packages/97/9f/da37cc4a188a1d5d203d65ab28d6504e17594b5342e0c1dc5610ee6f4535/numpy-1.21.6-cp37-cp37m-win_amd64.whl (14.0 MB)
Installing collected packages: numpy, torch
Successfully installed numpy-1.21.6 torch-1.1.0

到此這篇關於解決anaconda安裝pytorch報錯找不到包的問題的文章就介紹到這了,更多相關anaconda安裝pytorch報錯內容請搜尋it145.com以前的文章或繼續瀏覽下面的相關文章希望大家以後多多支援it145.com!


IT145.com E-mail:sddin#qq.com