天天看点

windows10 64位上安装pytorch

作者是在anaconda中虚拟环境中安装的

1. 首先安装anaconda, 参考https://www.jianshu.com/p/eaee1fadc1e9

2. 添加镜像

     conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/

     conda config --set show_channel_urls yes

3. anacondac创建一个虚拟环境,作者在这里设的是python3.6,进入控制台, 输入

    conda create -n pytorch python=3.6.2

4. 激活这个虚拟环境(取消激活用deactivate)

    activate pytorch

5. 采用pip安装cpu版本:

    pip install http://download.pytorch.org/whl/cpu/torch-0.4.0-cp36-cp36m-win_amd64.whl

测试torch是否安装成功

    import torch

    print(torch.__version__)

6. 安装torchvision

    pip install torchvision

继续阅读