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对应可用的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)
torch | torchvision | python |
---|---|---|
1.7.0 | 0.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.whl
为torch-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 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
找不到包,好难过,各种方法都报错。