pytorch安装超详细教程(Win10考虑CUDApython版本,含各种报错情况)

编程入门 行业动态 更新时间:2024-10-15 14:14:06

pytorch安装超详细教程(Win10考虑CUDApython版本,含各种<a href=https://www.elefans.com/category/jswz/34/1771188.html style=报错情况)"/>

pytorch安装超详细教程(Win10考虑CUDApython版本,含各种报错情况)

Step1:仔细地查看可以安装的torch版本

(47条消息) GPU版本pyTorch安装教程(实测,特别详细)_楊龘龘的博客-CSDN博客_安装gpu版本的pytorch要安装GPU版本的torch可以先参考此教程~

Windows 安装 CUDA/cuDNN - 知乎 (zhihu)

驱动程序版本为472.12,按照Release Notes :: CUDA Toolkit Documentation (nvidia),查得可使用CUDA 11.4.0 GA、CUDA 11.4 Update 1、CUDA 11.4 Update 2。

我的电脑驱动可适配的CUDA最高版本为11.4,电脑中安装有10.1和11.4

查看CUDA对应可用的torch版本,PyTorch和CUDA版本对应关系 - 墨莲玦 - 博客园 (cnblogs),以我的电脑为例,使用CUDA10.1,适配torch版本如下。

PyTorch 版本CUDA 环境
1.4.0、1.5.0(1)、1.6.0、1.7.0(1)10.1

查看python对应可用的torch版本,pytorch/vision: Datasets, Transforms and Models specific to Computer Vision (github)

torchtorchvisionpython
1.7.00.8.0>=3.6, <=3.8

Step2:安装

使用.whl文件安装

在这里下载torch-1.7.0和torchvision-0.8.0的.whl文件,在终端使用pip安装。

cd C:\Users\bulru\Downloads # 替换成.whl文件所在路径
pip install "torch-"
pip install "torchvision-"

安装时可能出现以下错误#ERROR: torch-1.7.0+cu101-cp38-cp38-win_amd64.whl is not a supported wheel on this platform.,这可能是因为系统默认的python版本问题,可以在终端使用python查看版本。

若想要修改,可以在系统变量中修改python路径顺序。

这里查看了一下各软件的版本,python为3.8.8,pip所在环境与默认python一致,CUDA为10.1,我的电脑是64位的。故安装的torch没有问题。

哭死,那为什么还报错。。。

修改.whl文件名

torch-1.7.0+cu101-cp38-cp38-win_amd64.whltorch-1.7.0+cu101-cp38-cp38-win.whl,还是会报错。

搜到一篇解决方案,在终端中查看

import setuptools.pep425tags   #根据自己pep425tags文件位置进行相应更改
print(setuptools.pep425tags.get_supported())

开始疯狂报错

最终在CMD中运行python -m pip debug --verbose,可以看到pip支持的版本中有cp38-cp38-win_amd64,我哭死。。。也不是文件命名的问题。。。

看着这些报错,我难过呀。。。

激活conda环境后再安装

尝试了一下激活conda环境后再安装,成功!

AttributeError: module ‘typing’ has no attribute ‘_ClassVar’

安装torchvision时报错AttributeError: module 'typing' has no attribute '_ClassVar',修改C:\Users\bulru\anaconda3\Lib\site-packages\dataclasses.py第550行中的'_ClassVar''ClassVar'

再次安装

使用pip安装

pip install torch==1.7.0+cu101 -i /
pip install torchvision==0.8.0+cu101 
pip install torchaudio==0.7.2 -i 
pip3 install torch==1.7.0+cu101 -i / --trusted-host pypi.douban

找不到包,好难过,各种方法都报错。

更多推荐

pytorch安装超详细教程(Win10考虑CUDApython版本,含各种报错情况)

本文发布于:2024-02-10 22:48:05,感谢您对本站的认可!
本文链接:https://www.elefans.com/category/jswz/34/1677749.html
版权声明:本站内容均来自互联网,仅供演示用,请勿用于商业和其他非法用途。如果侵犯了您的权益请与我们联系,我们将在24小时内删除。
本文标签:报错   版本   情况   教程   详细

发布评论

评论列表 (有 0 条评论)
草根站长

>www.elefans.com

编程频道|电子爱好者 - 技术资讯及电子产品介绍!