torch环境安装问题

安装pytorch中所出现的一点问题的解决思路

因为最近要应对毕业论文,课题方向大概是《基于机器学习的–识别》。所以需要学习一点深度学习的知识。但没有想到安装的过程居然如此曲折
最后的解决方式是:重装anaconda

笔者安装的环境是python3.9+anaconda4.11.0
CUDA的版本是10.2
起先安装的是11.*的版本,但是发现可能会出现与pytorch版本不兼容的问题,于是降到了10.2
anaconda最先选择的是清华镜像站里面的,此步骤巨坑,暂且放下不表
torch环境安装问题_第1张图片

conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch

起初出现了http[0.00.00.00]这样的问题【具体的无法复现了,意会就行】
没办法使用conda指令安装,另辟蹊径,找pip安装的方式
在pytorch官网选择适合的版本
一般要安装的有“torch”和“torchvision”,善用ctrl+F
文件后缀的cp35指的是python3.5版本,查找一个适合自己系统的版本
在anaconda prompt里切换到新建的环境,pip安装torch和torchvision。最先开始系统报错:
Index **** [http:………………网站] NOT FOUND …… on this platform.
具体的无法还原了,也忘记留相关记录了 。
创建一个新文件

cmd:python -m pip debug --verbose

torch环境安装问题_第2张图片

里面定义的whl文件的格式
不同的版本可能会有差异,修改成对应格式的文件名就好了
再用pip install 文件位置
成功安装

新建一个文件进行测试

import torch    # 如正常则静默
a = torch.Tensor([1.])    # 如正常则静默
a.cuda()    # 如正常则返回"tensor([ 1.], device='cuda:0')"
from torch.backends import cudnn # 如正常则静默
cudnn.is_acceptable(a.cuda())    # 如正常则返回 "True"

随便找的一个
运行显示
torch环境安装问题_第3张图片
最后一行,不是有效的Win32应用程序
解决方案是在终端里面切换conda的运行环境到64位

conda info

torch环境安装问题_第4张图片

set CONDA_FORCE_32BIT=0

这一步理想结果应该是:
torch环境安装问题_第5张图片
但是笔者经历了一晚上发现完全切换不了,于是推测是不是anaconda的版本有问题。指令升级了一下,发现已经是最新版本。
换个思路,有没有可能是清华镜像源的问题。
Anaconda官网下载位置
一路通过,不赘述了
然后按照pytorch官网提供的指令
torch环境安装问题_第6张图片
新建环境,输入
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
安装,一路通过畅通无阻
torch环境安装问题_第7张图片
代码测试
torch环境安装问题_第8张图片

顺利通过
折磨了我两天的torch环境安装问题,没想到竟然是清华镜像源的问题。重装即解决。

你可能感兴趣的:(pytorch,深度学习,python)