pytorch?cuda安裝報錯的解決方法
2022.12.8在win+cuda11.8下安裝最新Pytorch GPU版時遇到包不兼容的問題,該文記錄安裝的整個流程
一、按照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
報錯包不兼容,出現(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
報錯找不到包
PackagesNotFoundError: The following packages are not available from current channels: - cudatoolkit=11.7
推測原因可能是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é)
到此這篇關(guān)于pytorch cuda安裝報錯解決的文章就介紹到這了,更多相關(guān)pytorch cuda安裝報錯內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家!
相關(guān)文章
Django contrib auth authenticate函數(shù)源碼解析
這篇文章主要介紹了Django contrib auth authenticate函數(shù)源碼解析,文中通過示例代碼介紹的非常詳細(xì),對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值,需要的朋友可以參考下2020-11-11Caffe卷積神經(jīng)網(wǎng)絡(luò)視覺層Vision?Layers及參數(shù)詳解
這篇文章主要為大家介紹了Caffe卷積神經(jīng)網(wǎng)絡(luò)視覺層Vision?Layers及參數(shù)詳解,有需要的朋友可以借鑒參考下,希望能夠有所幫助,祝大家多多進(jìn)步,早日升職加薪2022-06-06Python創(chuàng)建模塊及模塊導(dǎo)入的方法
這篇文章主要介紹了Python創(chuàng)建模塊及模塊導(dǎo)入的方法,實例分析了模塊的定義、導(dǎo)入及模塊屬性的使用技巧,并附帶說明了包的概念與用法,需要的朋友可以參考下2015-05-05python利用thrift服務(wù)讀取hbase數(shù)據(jù)的方法
今天小編就為大家分享一篇python利用thrift服務(wù)讀取hbase數(shù)據(jù)的方法,具有很好的參考價值,希望對大家有所幫助。一起跟隨小編過來看看吧2018-12-12python解析中國天氣網(wǎng)的天氣數(shù)據(jù)
最近學(xué)習(xí)python 感覺這門腳本語言十分靈活 而且功能十分強(qiáng)大 尤其是他re庫用于正則匹配十分強(qiáng)大,寫了個例子解析中國天氣網(wǎng)2014-03-03python?yaml文件數(shù)據(jù)按原有的數(shù)據(jù)順序dump問題小結(jié)
這篇文章主要介紹了python?yaml文件數(shù)據(jù)按原有的數(shù)據(jù)順序dump,本文給大家介紹的非常詳細(xì),對大家的學(xué)習(xí)或工作具有一定的參考借鑒價值,需要的朋友參考下吧2023-11-11