torch gpu装了库里没有 torch装不上

转载

mob64ca1412b28c 2024-05-09 11:03:54

文章标签 torch gpu装了库里没有 pytorch 深度学习 python 官网 文章分类 游戏开发

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

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

笔者安装的环境是python3.9+anaconda4.11.0

CUDA的版本是10.2

起先安装的是11.*的版本，但是发现可能会出现与pytorch版本不兼容的问题，于是降到了10.2

anaconda最先选择的是清华镜像站里面的，此步骤巨坑，暂且放下不表

torch gpu装了库里没有 torch装不上_python

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 gpu装了库里没有 torch装不上_深度学习_02

里面定义的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 gpu装了库里没有 torch装不上_python_03