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

前言

使用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:\Anaconda3\envs\pytorch\Lib\site-packages\wheel>pip install torch-1.1.0-cp37-cp37m-win_amd64.whl
Looking in indexes: https://pypi.tuna.tsinghua.edu.cn/simple
Processing d:\anaconda3\envs\pytorch\lib\site-packages\wheel\torch-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報錯內容請搜索WalkonNet以前的文章或繼續瀏覽下面的相關文章希望大傢以後多多支持WalkonNet!

推薦閱讀: