zoukankan      html  css  js  c++  java
  • Pytorch离线安装方法

    诉求

    由于一些内网环境无法使用pip命令安装python三方库,寻求一种能够离线安装pytorch的方法。

    方法

    由于是内网,首选使用Anaconda代替Python,这样无需手动配置numpy等额外依赖。

    访问pytorch离线下载网址根据系统和CUDA版本选择自己需要的whl文件

    一共有两个,pytorchtorchvision,例如win10x64下cuda10.1的python3.7对应下载文件为

    cu101/torch-1.5.0%2Bcu101-cp37-cp37m-win_amd64.whl
    cu101/torchvision-0.6.0%2Bcu101-cp37-cp37m-win_amd64.whl
    

    源址下载较慢可用迅雷等P2P工具替代之

    之后是离线安装,打开Anaconda Navigator - Powershell Prompt输入

    (base) PS E:cu101> pip install .	orch-1.5.0+cu101-cp37-cp37m-win_amd64.whl .	orchvision-0.6.0+cu101-cp37-cp37m-win_amd64.whl
    

    注意顺序,后装的vision依赖torch,中途会自动搜寻并发现满足的numpy,future,pillow等依赖。

    验证

    至此,安装完毕,使用torch自带函数验证是否成功。

    >>> import torch
    >>> torch.cuda.is_available()
    True
    
  • 相关阅读:
    单例模式的八种写法
    反射
    工厂模式
    Java内存分配、管理小结
    Java 反射(二)
    Java反射
    servlet的web-xml配置详解
    substr和substring的区别
    C/C++中extern关键字详解
    lua总则
  • 原文地址:https://www.cnblogs.com/azureology/p/13041964.html
Copyright © 2011-2022 走看看