日本免费高清视频-国产福利视频导航-黄色在线播放国产-天天操天天操天天操天天操|www.shdianci.com

學(xué)無先后，達(dá)者為師

網(wǎng)站首頁 編程語言 正文

pytorch?cuda安裝報(bào)錯(cuò)的解決方法_python

作者：舒克兒不開飛機(jī) ? 更新時(shí)間： 2023-02-17 編程語言

2022.12.8在win+cuda11.8下安裝最新Pytorch GPU版時(shí)遇到包不兼容的問題，該文記錄安裝的整個(gè)流程

一、按照cuda版本在官網(wǎng)找命令

Pytorch官網(wǎng)命令如下

在win、cuda11.8下安裝GPU版Pytorch

conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia

分析該命令，-c后面為安裝包的源，-c pytorch表示使用官方源，可能出現(xiàn)速度慢等問題

二、換源

因此一般先將conda切換清華源

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/

三、Bug描述

換源后去掉-c執(zhí)行命令如下

conda install pytorch torchvision torchaudio pytorch-cuda=11.7

報(bào)錯(cuò)包不兼容，出現(xiàn)沖突

Found conflicts! Looking for incompatible packages.
This can take several minutes.  Press CTRL-C to abort.
Examining torchaudio:  43%|██████████████████████████████████████████████▎                                                             | 3/7 [00:22<00:24,  6.10s/it]/failed                                                                                                                                                                

UnsatisfiableError: The following specifications were found
to be incompatible with the existing python installation in your environment:

Specifications:

  - torchaudio -> python[version='2.7.*|3.5.*|3.6.*|>=2.7,<2.8.0a0|>=3.5,<3.6.0a0|3.4.*']

Your python: python=3.7

If python is on the left-most side of the chain, that's the version you've asked for.
When python appears to the right, that indicates that the thing on the left is somehow
not available for the python version you are constrained to. Note that conda will not
change your python version to a different minor version unless you explicitly specify
that.

The following specifications were found to be incompatible with each other:

Output in format: Requested package -> Available versions

Package typing conflicts for:
torchvision -> typing_extensions -> typing[version='>=3.7.4']
pytorch -> typing_extensions -> typing[version='>=3.7.4']

Package pytorch conflicts for:
torchaudio -> pytorch[version='1.10.0|1.10.1|1.10.2|1.11.0|1.12.0|1.12.1|1.13.0|1.9.1|1.9.0|1.8.1|1.8.0|1.7.1|1.7.0|1.6.0']
torchvision -> pytorch[version='1.10.0|1.10.1|1.10.2|1.11.0|1.12.0|1.12.1|1.13.0|1.9.1|1.9.0|1.8.1|1.8.0|1.7.1|1.7.0|1.6.0|1.5.1|1.5.0|1.4.0|1.3.1|1.3.0|1.2.0|>=1.1.0|>=1.0.0|>=0.4']

Package pytorch-cuda conflicts for:
pytorch -> pytorch-cuda[version='>=11.6,<11.7|>=11.7,<11.8']
torchaudio -> pytorch-cuda[version='11.6.*|11.7.*']
torchvision -> pytorch==1.13.0 -> pytorch-cuda[version='>=11.6,<11.7|>=11.7,<11.8']
torchaudio -> pytorch==1.13.0 -> pytorch-cuda[version='>=11.6,<11.7|>=11.7,<11.8']
torchvision -> pytorch-cuda[version='11.6.*|11.7.*']

Package requests conflicts for:
python=3.7 -> pip -> requests
torchvision -> request

四、解決方法

參考之前安裝成功的命令為

conda install pytorch torchvision torchaudio cudatoolkit=11.3

嘗試執(zhí)行

conda install pytorch torchvision torchaudio cudatoolkit=11.7

報(bào)錯(cuò)找不到包

PackagesNotFoundError: The following packages are not available from current channels:
  - cudatoolkit=11.7

推測(cè)原因可能是cuda11.7將cudatoolkit=11.7換為pytorch-cuda=11.7，而-c nvidia為pytorch-cuda的源。

執(zhí)行如下命令成功安裝

 conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c nvidia

五、查看是否使用GPU

import torch

use_gpu = torch.cuda.is_available()
print(use_gpu)
print("devices count:", torch.cuda.device_count())

輸出

True
devices count: 1

總結(jié)

原文鏈接：https://blog.csdn.net/JamesShuker/article/details/128257519

欄目分類
最近更新