当前位置:   article > 正文

Ubuntu18.04+Cuda10.1+Python3.6 下安装 PyTorch1.4.0+torchvision0.5.0,成功安装torch1.4.0和torchvision并解决安装速度过慢_cuda10.1安装torch

cuda10.1安装torch

1、添加镜像安装(失败了,)

在这里插入图片描述

2、使用whl文件分别安装(分别安装,成功了)

2.1 确定与当前Python3.6版本相匹配的Pytorch、torchvision细节信息

  • 如果不安装支持Python3.6的版本号,会出现错误:“ XXX is not a supported wheel on this platform”,因此,需要查询支持Python3.6的详细本版信息

    在这里插入图片描述

2.1.1 pip 20.0以下版本

  • 方式一:通过查询,我选择安装(cp36, cp36m, manylinux1_x86_64)

    cp35、cp36这样的标志,这就代表你要安装进的那个环境中的python版本(3.5、3.6)

    import pip
    print(pip.pep425tags.get_supported())
    
    • 1
    • 2
    在这里插入图片描述
  • 方式二
    import pip._internal
    print(pip._internal.pep425tags.get_supported())
    
    • 1
    • 2

2.2.2 pip 20.0及以上版本

  • 使用上述旧教程,会得到pip has no attribute pep425tags
    在这里插入图片描述
  • 直接在命令行中输入,不要在Python解释器输入以下命令:
    pip debug --verbose
    
    • 1
    从中我们可以看到Compatible tags字样,这些就是当前Python版本可以适配的标签。

参考链接(十分详细):https://blog.csdn.net/happywlg123/article/details/107281936

2.2 安装对应的Pytorch1.4.0

  • whl下载网址,下载合适的pytorch1.4(cp36, cp36m, manylinux1_x86_64
    在这里插入图片描述
  • pip list查一下你有没有wheel这个库,没有的话自己手动装一下
  • 和一般用pip安装类似,不过这次是直接用whl文件(方法为: pip install *.whl),进入pytorch的whl存放的目录,输入pip install torch-1.4.0-cp36-cp36m-linux_x86_64.whl进行安装
    在这里插入图片描述

2.3 安装对应的torchvision0.5.0

  • 要先安装PyTorch,再去安装torchvision,否则会报错
  • 这里查看与Pytorch1.4对应的torchvision的版本号
    在这里插入图片描述
  • 这里,下载对应的torchvision(cp36, cp36m, manylinux1_x86_64
    在这里插入图片描述
  • pip install torchvision-0.5.0-cp36-cp36m-manylinux1_x86_64.whl安装
    在这里插入图片描述

3、补充

  • 这个方法可以说是不仅仅适用于PyTorch的安装的,其他的任何的库都是可以的,当遇到无法直接下载的情况的时候,也可以尝试去下载它的whl,然后通过whl去安装我们的库。

Pytorch GPU版本whl文件安装

python安装torch踩过的坑

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

闽ICP备14008679号