当前位置:   article > 正文

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

win10安装pytorch

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

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

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

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

查看电脑驱动程序版本

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

电脑驱动可适配的最高CUDA版本

查看CUDA对应可用的torch版本,PyTorch和CUDA版本对应关系 - 墨莲玦 - 博客园 (cnblogs.com),以我的电脑为例,使用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.com)

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-"
  • 1
  • 2
  • 3

安装时可能出现以下错误#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())
  • 1
  • 2

开始疯狂报错
在这里插入图片描述
最终在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 https://download.pytorch.org/whl/torch/
pip install torchvision==0.8.0+cu101 
pip install torchaudio==0.7.2 -i https://download.pytorch.org/whl/torch_stable
pip3 install torch==1.7.0+cu101 -i http://pypi.douban.com/simple/ --trusted-host pypi.douban.com
  • 1
  • 2
  • 3
  • 4

在这里插入图片描述

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

本文内容由网友自发贡献,转载请注明出处:https://www.wpsshop.cn/w/IT小白/article/detail/810678
推荐阅读
相关标签
  

闽ICP备14008679号