Windows下用Anaconda安装Pytorch/torchvision

墨蓝 2022-06-01 05:46 518阅读 0赞

  1. 安装Anaconda最新版

  2. 依赖的环境

    • Anaconda3 x64 (with Python 3.5/3.6)
    • Windows 64位系统（Windows 7 或 Windows Server 2008 及以上）
    • CUDA 8 / CUDA 9
    • cuDNN v5以上
    • 如果安装了CUDA编译的包，请确保你的电脑有Nvidia的显卡。
    • 注：这里没有介绍GPU版本的安装方法，如需要的请搜索其他博文。
  1. 开始菜单打开Anaconda Prompt，在里面输入conda create -n pytorch python=3.5，为pytorch创建一个虚拟环境

3.5 添加清华镜像

conda config —addchannels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/

conda config —setshow_channel_urls yes

  1. activate pytorch 激活这个虚拟环境（取消激活用deactivate）

  2. 在这个环境中，

    for CPU only packages

    conda install -c peterjc123 pytorch

    for Windows 10 and Windows Server 2016, CUDA 8

    conda install -c peterjc123 pytorch cuda80

    for Windows 10 and Windows Server 2016, CUDA 9

    conda install -c peterjc123 pytorch cuda90

    for Windows 7/8/8.1 and Windows Server 2008/2012, CUDA 8

    conda install -c peterjc123 pytorch_legacy cuda80

5.5. torch安装好了其实还是需要torchvision的

使用本地安装的方法，先下载安装包

https://pypi.python.org/pypi/torchvision/0.1.8

然后在安装包所在的目录中用下面的命令安装:

pip install 下载的安装包的名字.whl

  1. 测试torch

    在Anaconda Prompt 里面打开python 然后键入import torch

    还有 import torchvision

  2. 测试CUDA和cuDNN

    CUDA TEST

    import torch
    x = torch.Tensor([1.0])
    xx = x.cuda()
    print(xx)

    CUDNN TEST

    from torch.backends import cudnn
    print(cudnn.is_acceptable(xx))

  3. 如果CUDA工作不正常，就要关掉了

    cudnn.enabled = False

  4. over

发表评论

表情：
评论列表 （有 0 条评论，518人围观）

还没有评论，来说两句吧...

相关阅读