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

  • 相关阅读:
    反射:框架设计的灵魂
    Junit测试
    XML笔记
    map 的用法
    opencv总结1
    光源
    镜面反射
    openGL纹理对象
    GPU入门
    动态规划1
  • 原文地址:https://www.cnblogs.com/jeshy/p/11778283.html
Copyright © 2011-2022 走看看