赞
踩
因为最近要应对毕业论文,课题方向大概是《基于机器学习的–识别》。所以需要学习一点深度学习的知识。但没有想到安装的过程居然如此曲折
最后的解决方式是:重装anaconda
笔者安装的环境是python3.9+anaconda4.11.0
CUDA的版本是10.2
起先安装的是11.*的版本,但是发现可能会出现与pytorch版本不兼容的问题,于是降到了10.2
anaconda最先选择的是清华镜像站里面的,此步骤巨坑,暂且放下不表
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
里面定义的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"
随便找的一个
运行显示
最后一行,不是有效的Win32应用程序
解决方案是在终端里面切换conda的运行环境到64位
conda info
set CONDA_FORCE_32BIT=0
这一步理想结果应该是:
但是笔者经历了一晚上发现完全切换不了,于是推测是不是anaconda的版本有问题。指令升级了一下,发现已经是最新版本。
换个思路,有没有可能是清华镜像源的问题。
Anaconda官网下载位置
一路通过,不赘述了
然后按照pytorch官网提供的指令
新建环境,输入
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
安装,一路通过畅通无阻
代码测试
顺利通过
折磨了我两天的torch环境安装问题,没想到竟然是清华镜像源的问题。重装即解决。
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。