zoukankan      html  css  js  c++  java
  • .whl文件安装cuda10.0版本的pytorch1.3.0+torchvision0.4.1

    $ python3 -m venv env3
    $ source env3/bin/activate
    $ cd env3/share/python-wheels

    在此找到对应的版本:https://download.pytorch.org/whl/cu100/torch_stable.html
    $ wget https://download.pytorch.org/whl/cu100/torch-1.3.0%2Bcu100-cp36-cp36m-linux_x86_64.whl
    $ wget https://download.pytorch.org/whl/cu100/torchvision-0.4.1%2Bcu100-cp36-cp36m-linux_x86_64.whl
    $ pip install torch-1.3.0+cu100-cp36-cp36m-linux_x86_64.whl torchvision-0.4.1+cu100-cp36-cp36m-linux_x86_64.whl
    >>>>>Processing ./torch-1.3.0+cu100-cp36-cp36m-linux_x86_64.whl
    >>>>>Processing ./torchvision-0.4.1+cu100-cp36-cp36m-linux_x86_64.whl
    >>>>>Collecting numpy (from torch==1.3.0+cu100)
    >>>>>...................................................................
    >>>>>Successfully installed numpy-1.17.3 pillow-6.2.1 six-1.12.0 torch-1.3.0+cu100 torchvision-0.4.1+cu100
    $ python
    $ import torch
    >>> print(torch.version.cuda)
    10.0.1

  • 相关阅读:
    Linux下动态库生成和使用
    ELK日志平台
    zabbix 监控端口
    zabbix 分布式zabbix_proxy
    zabbix 邮件报警 监控mysql主从
    Kubernetes 本地仓库
    kubernetes ui 搭建
    Kubernetes K8s
    Dockerfile
    Docker 指定容量
  • 原文地址:https://www.cnblogs.com/jeshy/p/11778283.html
Copyright © 2011-2022 走看看